pytorch_pfn_extras.handler.BaseLogic¶
- class pytorch_pfn_extras.handler.BaseLogic(options=None)¶
- Parameters
options (Optional[Dict[str, Any]]) –
- __init__(options=None)¶
- Parameters
options (Optional[Dict[str, Any]]) –
Methods
__init__([options])consume_options(options)A method to update options of Logic.
eval_step(models, batch_idx, batch)A method for an evaluation step.
train_epoch_begin(models, epoch, loader)A method called when starting a new epoch of training.
train_epoch_end(models, epoch)A method called when completing an epoch of training.
train_step(models, optimizers, batch_idx, batch)A method invokes the models forward and backward passes.
train_step_optimizers(models, optimizers, …)A method in charge of stepping the provided optimizers.
train_validation_begin(models)A method called when starting a validation.
train_validation_end(models)A method called when the validation completes.