Restore train_epochs to 200 instead of 8

2021-08-05 18:32:46 +02:00
parent 40c55478f3
commit 8bce5d15f6
2 changed files with 3 additions and 2 deletions
--- a/config/value_dice.json5
+++ b/config/value_dice.json5
@@ -78,7 +78,8 @@
            lr: 1e-3,
            weight_decay: 0.1,
    },
-    train_epochs: 8,
+    train_epochs: 200,
    train_batch_size: 32,
    discount: 0.95,
+    clip_grad_norm: 1.,
 }