pytorch_pfn_extras.writing#

Classes

pytorch_pfn_extras.writing.ProcessQueueWriter([...])

Snapshot writer that uses process queue.

pytorch_pfn_extras.writing.ProcessWriter([...])

Snapshot writer that uses a separate process.

pytorch_pfn_extras.writing.QueueWriter([...])

Base class of queue snapshot writers.

pytorch_pfn_extras.writing.SimpleWriter([...])

The most simple snapshot writer.

pytorch_pfn_extras.writing.StandardWriter([...])

Base class of snapshot writers which use thread or process.

pytorch_pfn_extras.writing.TensorBoardWriter([...])

Writer that sends statistics to TensorBoard.

pytorch_pfn_extras.writing.ThreadQueueWriter([...])

Snapshot writer that uses a thread queue.

pytorch_pfn_extras.writing.ThreadWriter([...])

Snapshot writer that uses a separate thread.

pytorch_pfn_extras.writing.Writer([fs, out_dir])

Base class of snapshot writers.