pytorch_pfn_extras.training.extensions.snapshot_writers#

Classes

pytorch_pfn_extras.training.extensions.snapshot_writers.ProcessQueueWriter([...])

Snapshot writer that uses process queue.

pytorch_pfn_extras.training.extensions.snapshot_writers.ProcessWriter([...])

Snapshot writer that uses a separate process.

pytorch_pfn_extras.training.extensions.snapshot_writers.QueueWriter([...])

Base class of queue snapshot writers.

pytorch_pfn_extras.training.extensions.snapshot_writers.SimpleWriter([...])

The most simple snapshot writer.

pytorch_pfn_extras.training.extensions.snapshot_writers.StandardWriter([...])

Base class of snapshot writers which use thread or process.

pytorch_pfn_extras.training.extensions.snapshot_writers.TensorBoardWriter([...])

Writer that sends statistics to TensorBoard.

pytorch_pfn_extras.training.extensions.snapshot_writers.ThreadQueueWriter([...])

Snapshot writer that uses a thread queue.

pytorch_pfn_extras.training.extensions.snapshot_writers.ThreadWriter([...])

Snapshot writer that uses a separate thread.

pytorch_pfn_extras.training.extensions.snapshot_writers.Writer([...])

Base class of snapshot writers.