pytorch_pfn_extras.training.extensions.SnapshotMode

class pytorch_pfn_extras.training.extensions.SnapshotMode(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: Enum

Attributes

DEFAULT
DISTRIBUTED
SHARDED

DEFAULT = 1

DISTRIBUTED = 2

SHARDED = 3