vak.train.parametric_umap.get_trainer

vak.train.parametric_umap.get_trainer(accelerator: str, devices: int | list[int], max_epochs: int, ckpt_root: str | Path, ckpt_step: int, log_save_dir: str | Path) Trainer[source]

Returns an instance of lightning.pytorch.Trainer with a default set of callbacks. Used by vak.core functions.