mem maze env ok 1.2

2023-06-18 09:16:32 +08:00
parent 152415f32e
commit 8e005afde5
5 changed files with 15 additions and 49 deletions
--- a/configs.yaml
+++ b/configs.yaml
@@ -156,8 +156,10 @@ debug:
  batch_size: 10
  batch_length: 20

-mazegym:
-  task: '9'
+MemoryMaze:
+  actor_dist: 'onehot'
+  imag_gradient: 'reinforce'
+  task: '9x9'
  steps: 1e6
  action_repeat: 2