如何监控Chainer框架中验证集的错误?

时间:2017-08-11 19:14:01

标签: neural-network chainer

我是Chainer的新手,并编写了一个训练简单的前馈神经网络的代码。我有一个验证集和一个列车集,并希望测试每个类似500次迭代的验证集,如果结果更好,我想保存我的网络权重。谁能告诉我怎么办?

这是我的代码:

optimizer = optimizers.Adam()
optimizer.setup(model)

updater = training.StandardUpdater(train_iter, optimizer, device=0)
trainer = training.Trainer(updater, (10000, 'epoch'), out='result')

trainer.extend(extensions.Evaluator(validation_iter, model, device=0))
trainer.extend(extensions.LogReport())
trainer.extend(extensions.PrintReport(['epoch', 'main/loss',  'validation/main/loss', 'elapsed_time']))
trainer.run()

0 个答案:

没有答案