chainer.training.extensions.snapshot_writers.ProcessQueueWriter¶
-
class
chainer.training.extensions.snapshot_writers.
ProcessQueueWriter
(savefun=<function save_npz>, task=None)[source]¶ Snapshot writer that uses process queue.
This class creates a process and a queue by
multiprocessing
module. The process will be a consumer of this queue, and the main process will be a producer of this queue.Note
Forking a new process from MPI process might be danger. Consider using
ThreadQueueWriter
instead ofProcessQueueWriter
if you are using MPI.Methods
-
__call__
(filename, outdir, target)[source]¶ Invokes the actual snapshot function.
This method is invoked by a
Snapshot
object every time it takes a snapshot.
-