vak.train.parametric_umap.get_trainer#

vak.train.parametric_umap.get_trainer(max_epochs: int, ckpt_root: str | Path, ckpt_step: int, log_save_dir: str | Path, device: str = 'cuda') Trainer[source]#

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