vak.train.frame_classification.get_trainer

vak.train.frame_classification.get_trainer(accelerator: str, devices: int | list[int], max_steps: int, log_save_dir: str | Path, val_step: int, callback_kwargs: dict | None = None) → Trainer

Returns an instance of lightning.pytorch.Trainer with a default set of callbacks.

Used by vak.train.frame_classification(). The default set of callbacks is provided by get_default_train_callbacks().

Parameters:

• accelerator (str)

• devices (int, list of int)

• max_steps (int)

• log_save_dir (str, pathlib.Path)

• val_step (int)

• default_callback_kwargs (dict, optional)

Returns:

trainer

Return type:

lightning.pytorch.Trainer