workflow.ignite package¶
-
workflow.ignite.
trainer
(train_batch, evaluate_batch, evaluate_data_loaders, metrics, optimizers)[source]¶ Create standard trainer with evaluators.
- train_batchfunction
function that trains on given batch
- evaluate_batchfunction
function that evaluates a given batch
- evaluate_data_loaders: list
data loaders that yield batches to evaluate on
- metricsdict
dict with one dict each for ‘train’ and evaluate data loader. Wrap a metric with trainer.Progress to show in progress bar.
- optimizersdict
dict with optimizers for logging
- tuple
trainer engine list of evaluator engines tensorboard logger