core.Baseline.Trainer

Module Contents

class core.Baseline.Trainer.Trainer(model, dataloaders, optimizer, scheduler=None, criterion=None, num_epochs=100, device='cuda:0', no_valid=False, valid_frequency=1, amp=False, algo_params=None)[source]

Bases: core.BaseTrainer.BaseTrainer

Abstract base class for trainer implementations

criterion[source]