fusionlab.trainers.trainer module# class fusionlab.trainers.trainer.Trainer(device)[source]# Bases: object fit(model, train_dataloader, val_dataloader, epochs, optimizer, loss_fn)[source]# on_epoch_begin()[source]# on_epoch_end()[source]# on_fit_begin()[source]# on_fit_end()[source]# train_epoch()[source]# train_step(data)[source]# val_epoch()[source]# val_step(data)[source]#