pytorch_pfn_extras.engine#

Functions

pytorch_pfn_extras.engine.cast(typ, val)

Cast a value to a type.

pytorch_pfn_extras.engine.create_evaluator(...)

Creates an evaluator object.

pytorch_pfn_extras.engine.create_trainer(...)

Creates a trainer object.

pytorch_pfn_extras.engine.default_transform_model(n, x)

pytorch_pfn_extras.engine.filter_state_objects(args)

pytorch_pfn_extras.engine.filter_state_objects_dict(args)

Classes

pytorch_pfn_extras.engine.DistributedEvaluator(...)

pytorch_pfn_extras.engine.Evaluator(handler, ...)

pytorch_pfn_extras.engine.NamedTuple(typename)

Typed version of namedtuple.

pytorch_pfn_extras.engine.StateObjectProtocol(...)

pytorch_pfn_extras.engine.Trainer(handler, ...)