hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated results

Browse Source
This commit is contained in:
NM512 2024-03-11 06:22:09 +09:00
parent 2cfcaefea2
commit 4e50f302cd
4 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -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)

BIN
imgs/dmcproprio.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 831 KiB

BIN
imgs/dmcvision.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 851 KiB

1
networks.py
View File

@ -145,7 +145,6 @@ class RSSM(nn.Module):
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]