BaseTrainer

class hy2dl.training.basetrainer.BaseTrainer(cfg: Config, training_dataset: BaseDataset)

Bases: object

Class to train a model

Parameters:

• cfg (Config) – Configuration object containing model hyperparameters and settings.

• training_dataset (hy2dl.datasetzoo.basedataset.BaseDataset) – Dataset used for training

train_model(epoch)