Merge pull request #236 from huangshiyu13/main

update readme
OpenRL-Lab · Oct 10, 2023 · 7403260 · 7403260
2 parents e2ea971 + 9ea9704
commit 7403260
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 1 deletion.
diff --git a/Gallery.md b/Gallery.md
@@ -61,6 +61,7 @@ Users are also welcome to contribute their own training examples and demos to th
 |                                    [CartPole](https://gymnasium.farama.org/environments/classic_control/cart_pole/)<br>  <img width="300px" height="auto" src="./docs/images/cartpole.png">                                    |                                                              ![discrete](https://img.shields.io/badge/-discrete-brightgreen)                                                              |      [code](./examples/cartpole/)       |
 |                            [MPE: Simple Spread](https://pettingzoo.farama.org/environments/mpe/simple_spread/)<br>  <img width="300px" height="auto" src="./docs/images/simple_spread_trained.gif">                            |                                    ![discrete](https://img.shields.io/badge/-discrete-brightgreen)  ![MARL](https://img.shields.io/badge/-MARL-yellow)                                    |         [code](./examples/mpe/)         |
 |                                                      [StarCraft II](https://github.com/oxwhirl/smac)<br>  <img width="300px" height="auto" src="./docs/images/smac.png">                                                       |                                    ![discrete](https://img.shields.io/badge/-discrete-brightgreen)  ![MARL](https://img.shields.io/badge/-MARL-yellow)                                    |        [code](./examples/smac/)         |
+|                                                        [SMACv2](https://github.com/oxwhirl/smacv2)<br>  <img width="300px" height="auto" src="https://github.com/oxwhirl/smacv2/raw/main/docs/imgs/reflect.png">                                                         |                                    ![discrete](https://img.shields.io/badge/-discrete-brightgreen)  ![MARL](https://img.shields.io/badge/-MARL-yellow)                                    |       [code](./examples/smacv2/)        |
 |                                    [Chat Bot](https://openrl-docs.readthedocs.io/en/latest/quick_start/train_nlp.html)<br>  <img width="300px" height="auto" src="./docs/images/chat.gif">                                     | ![discrete](https://img.shields.io/badge/-discrete-brightgreen)        ![NLP](https://img.shields.io/badge/-NLP-green)     ![Transformer](https://img.shields.io/badge/-Transformer-blue) |         [code](./examples/nlp/)         |
 |                                            [Atari Pong](https://gymnasium.farama.org/environments/atari/pong/)<br>  <img width="300px" height="auto" src="./docs/images/pong.png">                                             |                                 ![discrete](https://img.shields.io/badge/-discrete-brightgreen)        ![image](https://img.shields.io/badge/-image-red)                                  |        [code](./examples/atari/)        |
 |                              [PettingZoo: Tic-Tac-Toe](https://pettingzoo.farama.org/environments/classic/tictactoe/)<br>  <img width="300px" height="auto" src="./docs/images/tic-tac-toe.jpeg">                              |                                ![selfplay](https://img.shields.io/badge/-selfplay-blue)    ![discrete](https://img.shields.io/badge/-discrete-brightgreen)                                |      [code](./examples/selfplay/)       |

diff --git a/README.md b/README.md
@@ -115,6 +115,7 @@ Environments currently supported by OpenRL (for more details, please refer to [G
 - [Chat Bot](https://openrl-docs.readthedocs.io/en/latest/quick_start/train_nlp.html)
 - [Atari](https://gymnasium.farama.org/environments/atari/)
 - [StarCraft II](https://github.com/oxwhirl/smac)
+- [SMACv2](https://github.com/oxwhirl/smacv2)
 - [Omniverse Isaac Gym](https://github.com/NVIDIA-Omniverse/OmniIsaacGymEnvs)
 - [DeepMind Control](https://shimmy.farama.org/environments/dm_control/)
 - [Snake](http://www.jidiai.cn/env_detail?envid=1)

diff --git a/README_zh.md b/README_zh.md
@@ -91,6 +91,7 @@ OpenRL目前支持的环境（更多详情请参考 [Gallery](Gallery.md)）：
 - [Chat Bot](https://openrl-docs.readthedocs.io/en/latest/quick_start/train_nlp.html)
 - [Atari](https://gymnasium.farama.org/environments/atari/)
 - [StarCraft II](https://github.com/oxwhirl/smac)
+- [SMACv2](https://github.com/oxwhirl/smacv2)
 - [Omniverse Isaac Gym](https://github.com/NVIDIA-Omniverse/OmniIsaacGymEnvs)
 - [DeepMind Control](https://shimmy.farama.org/environments/dm_control/)
 - [Snake](http://www.jidiai.cn/env_detail?envid=1)

diff --git a/openrl/modules/networks/vdn_network.py b/openrl/modules/networks/vdn_network.py
@@ -194,7 +194,7 @@ def eval_values(self, obs, rnn_states, masks, action_masks=None):
                 obs[key] = check(obs[key]).to(**self.tpdv)
         else:
             if isinstance(obs, dict):
-              obs = obs["policy"]
+                obs = obs["policy"]
             obs = check(obs).to(**self.tpdv)
         rnn_states = check(rnn_states).to(**self.tpdv)
         masks = check(masks).to(**self.tpdv)