From 8bce5d15f6e2f6dbc87c13c8967f5c197a16121c Mon Sep 17 00:00:00 2001
From: Johannes Fischer <johannes.fischer@kit.edu>
Date: Thu, 5 Aug 2021 18:32:46 +0200
Subject: [PATCH] Restore train_epochs to 200 instead of 8

---
 config/networks.json5   | 2 +-
 config/value_dice.json5 | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

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