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.
-
finalize
()[source]¶ Finalizes the wirter.
Like extensions in
Trainer
, this method is invoked at the end of the training.
-
__eq__
()¶ Return self==value.
-
__ne__
()¶ Return self!=value.
-
__lt__
()¶ Return self<value.
-
__le__
()¶ Return self<=value.
-
__gt__
()¶ Return self>value.
-
__ge__
()¶ Return self>=value.
-