我是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()