zjowowen commited on
Commit
8f66718
·
1 Parent(s): 98bfb58

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -21,7 +21,7 @@ model-index:
21
  type: CartPole-v0
22
  metrics:
23
  - type: mean_reward
24
- value: 146.4 +/- 5.33
25
  name: mean_reward
26
  ---
27
 
@@ -29,11 +29,11 @@ model-index:
29
 
30
  ## Model Description
31
  <!-- Provide a longer summary of what this model is. -->
32
- This is a simple **MuZero** implementation to OpenAI/Gym/Box2d **CartPole-v0** by using [DI-engine](https://github.com/opendilab/di-engine) and [DI-zoo](https://github.com/opendilab/DI-engine/tree/main/dizoo).
33
 
34
  **DI-engine** is a python library for solving general decision intelligence problems, which is based on implementations of reinforcement learning framework using PyTorch or JAX. This library aims to standardize the reinforcement learning framework across different algorithms, benchmarks, environments, and to support both academic researches and prototype applications. Besides, self-customized training pipelines and applications are supported by reusing different abstraction levels of DI-engine reinforcement learning framework.
35
 
36
-
37
 
38
  ## Model Usage
39
  ### Install the Dependencies
@@ -142,6 +142,8 @@ push_model_to_hub(
142
  usage_file_by_huggingface_ding="./muzero/cartpole_muzero_download.py",
143
  train_file="./muzero/cartpole_muzero.py",
144
  repo_id="OpenDILabCommunity/CartPole-v0-MuZero",
 
 
145
  create_repo=False
146
  )
147
 
 
21
  type: CartPole-v0
22
  metrics:
23
  - type: mean_reward
24
+ value: 194.3 +/- 7.42
25
  name: mean_reward
26
  ---
27
 
 
29
 
30
  ## Model Description
31
  <!-- Provide a longer summary of what this model is. -->
32
+ This is a simple **MuZero** implementation to OpenAI/Gym/Box2d **CartPole-v0** by using [DI-engine](https://github.com/opendilab/di-engine) and [LightZero](https://github.com/opendilab/LightZero).
33
 
34
  **DI-engine** is a python library for solving general decision intelligence problems, which is based on implementations of reinforcement learning framework using PyTorch or JAX. This library aims to standardize the reinforcement learning framework across different algorithms, benchmarks, environments, and to support both academic researches and prototype applications. Besides, self-customized training pipelines and applications are supported by reusing different abstraction levels of DI-engine reinforcement learning framework.
35
 
36
+ **LightZero** is a lightweight, efficient, and easy-to-understand open-source algorithm toolkit that combines Monte Carlo Tree Search (MCTS) and Deep Reinforcement Learning (RL).
37
 
38
  ## Model Usage
39
  ### Install the Dependencies
 
142
  usage_file_by_huggingface_ding="./muzero/cartpole_muzero_download.py",
143
  train_file="./muzero/cartpole_muzero.py",
144
  repo_id="OpenDILabCommunity/CartPole-v0-MuZero",
145
+ platform_info="[DI-engine](https://github.com/opendilab/di-engine) and [LightZero](https://github.com/opendilab/LightZero)",
146
+ model_description="**LightZero** is a lightweight, efficient, and easy-to-understand open-source algorithm toolkit that combines Monte Carlo Tree Search (MCTS) and Deep Reinforcement Learning (RL).",
147
  create_repo=False
148
  )
149