Implement ValueDICE and some restructuring

2021-08-04 19:34:49 +02:00
parent cba42c6e4d
commit 5c40de66fa
8 changed files with 442 additions and 86 deletions
--- a/config/networks.json5
+++ b/config/networks.json5
@@ -36,7 +36,7 @@
            lr: 1e-3,
            weight_decay: 0.1,
    },
-    train_epochs: 200,
+    train_epochs: 8,
    train_batch_size: 32,
    loss: 'huber',
 }