Skip to content

Commit

Permalink
Write setup code for code example
Browse files Browse the repository at this point in the history
  • Loading branch information
niboshi committed Mar 26, 2018
1 parent b158939 commit 9c9d9c3
Showing 1 changed file with 14 additions and 1 deletion.
15 changes: 14 additions & 1 deletion chainer/training/extensions/_snapshot.py
Expand Up @@ -92,6 +92,19 @@ def snapshot(savefun=None,
Returns:
Snapshot extension object.
.. testcode::
:hide:
from chainer import training
class Model(chainer.Link):
def __call__(self, x):
return x
train_iter = chainer.iterators.SerialIterator([], 1)
optimizer = optimizers.SGD().setup(Model())
updater = training.updaters.StandardUpdater(
train_iter, optimizer, device=0)
trainer = training.Trainer(updater)
.. admonition:: Using asynchronous writers
By specifying ``writer`` argument, writing operations can be made
Expand All @@ -107,7 +120,7 @@ def snapshot(savefun=None,
>>> from chainer.training import extensions
>>> writer = extensions.snapshot_writers.ProcessWriter(
>>> savefun=extensions.snapshots.util.save_npz)
... savefun=extensions.snapshots.util.save_npz)
>>> trainer.extend(extensions.snapshot(writer=writer), \
trigger=(1, 'epoch'))
Expand Down

0 comments on commit 9c9d9c3

Please sign in to comment.