updated results

NM512 · Mar 10, 2024 · 4e50f30 · 4e50f30
1 parent 2cfcaef
commit 4e50f30
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -34,9 +34,9 @@ So far, the following benchmarks can be used for testing.
 
 ## Results
 #### DMC Proprio
-![dmcproprio](https://github.com/NM512/dreamerv3-torch/assets/70328564/0d3f4d44-d487-4097-bed1-3f0573aaf902)
+![dmcproprio](imgs/dmcproprio.png)
 #### DMC Vision
-![dmcvision](https://github.com/NM512/dreamerv3-torch/assets/70328564/162c24d0-5f06-4ef6-bbc3-202221f6ad11)
+![dmcvision](imgs/dmcvision.png)
 #### Atari 100k
 ![atari100k](https://github.com/NM512/dreamerv3-torch/assets/70328564/0da6d899-d91d-44b4-a8c4-d5b37413aa11)
 

diff --git a/imgs/dmcproprio.png b/imgs/dmcproprio.png
diff --git a/imgs/dmcvision.png b/imgs/dmcvision.png
diff --git a/networks.py b/networks.py
@@ -145,7 +145,6 @@ def observe(self, embed, action, is_first, state=None):
     def imagine_with_action(self, action, state):
         swap = lambda x: x.permute([1, 0] + list(range(2, len(x.shape))))
         assert isinstance(state, dict), state
-        action = action
         action = swap(action)
         prior = tools.static_scan(self.img_step, [action], state)
         prior = prior[0]