diff --git a/config/networks.json5 b/config/networks.json5
index ac10d2f..ebce18f 100644
--- 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',
 }
\ No newline at end of file
diff --git a/config/value_dice.json5 b/config/value_dice.json5
index 0952bd1..ba12c15 100644
--- 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.,
 }
\ No newline at end of file