core.BaseTrainer ================ .. py:module:: core.BaseTrainer Module Contents --------------- .. py:class:: BaseTrainer(model, dataloaders, optimizer, scheduler=None, criterion=None, num_epochs=100, device='cuda:0', no_valid=False, valid_frequency=1, amp=False, algo_params=None) Abstract base class for trainer implementations .. py:attribute:: model .. py:attribute:: dataloaders .. py:attribute:: optimizer .. py:attribute:: scheduler :value: None .. py:attribute:: criterion :value: None .. py:attribute:: num_epochs :value: 100 .. py:attribute:: no_valid :value: False .. py:attribute:: valid_frequency :value: 1 .. py:attribute:: device :value: 'cuda:0' .. py:attribute:: amp :value: False .. py:attribute:: best_weights :value: None .. py:attribute:: best_f1_score :value: 0.1 .. py:attribute:: scaler :value: None .. py:attribute:: loss_metric .. py:attribute:: f1_metric .. py:attribute:: post_pred .. py:attribute:: post_gt .. py:method:: train() Train the model