QueueWriter(savefun=<function save_npz>, task=None)¶
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.
savefun – Callable object which is passed to the
create_task()if the task is
None. 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 to
Writer.__call__. This object is directly put into the queue.
__call__(filename, outdir, target)¶
Invokes the actual snapshot function.
This method is invoked by a
Snapshotobject every time it takes a snapshot.
Finalizes the wirter.
Like extensions in
Trainer, this method is invoked at the end of the training.
save(filename, outdir, target, savefun, **kwds)¶