Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -21,7 +21,7 @@ model-index:
|
|
| 21 |
type: CartPole-v0
|
| 22 |
metrics:
|
| 23 |
- type: mean_reward
|
| 24 |
-
value:
|
| 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 [
|
| 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 |
|