chainer.training.extensions.snapshot_writers.QueueWriter¶
- class chainer.training.extensions.snapshot_writers.QueueWriter(savefun=<function save_npz>, task=None)[source]¶
Base class of queue snapshot writers.
This class is a base class of snapshot writers that use a queue. A Queue is created when this class is constructed, and every time when
__call__
is invoked, a snapshot task is put into the queue.- Parameters
savefun – Callable object which is passed to the
create_task()
if the task isNone
. It takes three arguments: the output file path, the serialized dictionary object, and the optional keyword arguments.task – Callable object. Its
__call__
must have a same interface toWriter.__call__
. This object is directly put into the queue.
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__(value, /)¶
Return self==value.
- __ne__(value, /)¶
Return self!=value.
- __lt__(value, /)¶
Return self<value.
- __le__(value, /)¶
Return self<=value.
- __gt__(value, /)¶
Return self>value.
- __ge__(value, /)¶
Return self>=value.