pytorch_pfn_extras.training#

Functions

pytorch_pfn_extras.training.make_extension([...])

Decorator to make given function into an extension.

Classes

pytorch_pfn_extras.training.AccuracyMetric(...)

A metric for an evaluator to report accuracy.

pytorch_pfn_extras.training.DistributedEvaluator(...)

pytorch_pfn_extras.training.Evaluator(...[, ...])

pytorch_pfn_extras.training.Extension()

Base class of extensions.

pytorch_pfn_extras.training.ExtensionEntry(...)

Extension and options.

pytorch_pfn_extras.training.ExtensionsManager(...)

Manages the extensions and the current status.

pytorch_pfn_extras.training.ExtensionsManagerProtocol(...)

pytorch_pfn_extras.training.IgniteExtensionsManager(...)

Manages extensions and the current status in Ignite training loop.

pytorch_pfn_extras.training.StateObjectProtocol(...)

pytorch_pfn_extras.training.Trainer(handler, ...)

Modules

pytorch_pfn_extras.training.extension

pytorch_pfn_extras.training.extensions

pytorch_pfn_extras.training.manager

pytorch_pfn_extras.training.metrics

pytorch_pfn_extras.training.trigger

pytorch_pfn_extras.training.triggers