Add arg for fit() method

seiichiinoue · Jun 19, 2018 · 4e0dd07 · 4e0dd07
1 parent 17faf16
commit 4e0dd07
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ch06/train_rnnlm.py b/ch06/train_rnnlm.py
@@ -30,7 +30,8 @@
 trainer = RnnlmTrainer(model, optimizer)
 
 # 勾配クリッピングを適用して学習
-trainer.fit(xs, ts, max_epoch, batch_size, time_size, max_grad)
+trainer.fit(xs, ts, max_epoch, batch_size, time_size, max_grad,
+            eval_interval=20)
 trainer.plot(ylim=(0, 500))
 
 # テストデータで評価