train¶
- trojanzoo.utils.train.train(module, num_classes, epochs, optimizer, lr_scheduler=None, lr_warmup_epochs=0, model_ema=None, model_ema_steps=32, grad_clip=None, pre_conditioner=None, print_prefix='Train', start_epoch=0, resume=0, validate_interval=10, save=False, amp=False, loader_train=None, loader_valid=None, epoch_fn=None, get_data_fn=None, forward_fn=None, loss_fn=None, after_loss_fn=None, validate_fn=None, save_fn=None, file_path=None, folder_path=None, suffix=None, writer=None, main_tag='train', tag='', accuracy_fn=None, verbose=True, output_freq='iter', indent=0, change_train_eval=True, lr_scheduler_freq='epoch', backward_and_step=True, **kwargs)[source]¶
Train the model
- trojanzoo.utils.train.validate(module, num_classes, loader, print_prefix='Validate', indent=0, verbose=True, get_data_fn=None, forward_fn=None, loss_fn=None, writer=None, main_tag='valid', tag='', _epoch=None, accuracy_fn=None, **kwargs)[source]¶
Evaluate the model.
- Returns:
(float, float) – Accuracy and loss.