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/networks.json5
+++ b/config/networks.json5
@@ -36,7 +36,7 @@
            lr: 1e-3,
            weight_decay: 0.1,
    },
-    train_epochs: 8,
+    train_epochs: 200,
    train_batch_size: 32,
    loss: 'huber',
 }