nfflr.train.trainer.setup_trainer

Contents

nfflr.train.trainer.setup_trainer#

nfflr.train.trainer.setup_trainer(model, criterion, optimizer, scheduler, prepare_batch, config: TrainingConfig)[source]#

Create ignite trainer and attach common event handlers.