Pyramids model
Browse files- Pyramids.onnx +3 -0
- README.md +29 -0
- config.json +1 -0
- configuration.yaml +90 -0
Pyramids.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:33590ede21dd0083b79de4702a2fb42aab48fa0ceac5a818fb217683b9623ca4
|
| 3 |
+
size 1418184
|
README.md
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: ml-agents
|
| 3 |
+
tags:
|
| 4 |
+
- Pyramids
|
| 5 |
+
- deep-reinforcement-learning
|
| 6 |
+
- reinforcement-learning
|
| 7 |
+
- ML-Agents-Pyramids
|
| 8 |
+
- curiosity
|
| 9 |
+
- RND
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# **ppo** Agent playing **Pyramids**
|
| 13 |
+
This is a trained model of a **ppo** agent playing **Pyramids**
|
| 14 |
+
using the [Unity ML-Agents Library](https://github.com/Unity-Technologies/ml-agents).
|
| 15 |
+
|
| 16 |
+
This agent uses **Random Network Distillation (RND)** for curiosity-driven exploration to solve the Pyramids environment, where it must press a button to spawn a pyramid, navigate to it, knock it over, and reach the gold brick at the top.
|
| 17 |
+
|
| 18 |
+
## Usage (with ML-Agents)
|
| 19 |
+
The Documentation: https://unity-technologies.github.io/ml-agents/ML-Agents-Toolkit-Documentation/
|
| 20 |
+
|
| 21 |
+
We wrote a complete tutorial to learn to train your first agent using ML-Agents and publish it to the Hub:
|
| 22 |
+
- A *short tutorial* where you teach Huggy the Dog to fetch the stick and then play with him directly in your
|
| 23 |
+
browser: https://huggingface.co/learn/deep-rl-course/unitbonus1/introduction
|
| 24 |
+
- A *longer tutorial* to understand how works ML-Agents:
|
| 25 |
+
https://huggingface.co/learn/deep-rl-course/unit5/introduction
|
| 26 |
+
|
| 27 |
+
### Resume the training
|
| 28 |
+
```bash
|
| 29 |
+
mlagents-learn <your_configuration_file_path.yaml> --run-id=<run_id> --resume
|
config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"default_settings": null, "behaviors": {"Pyramids": {"trainer_type": "ppo", "hyperparameters": {"batch_size": 128, "buffer_size": 2048, "learning_rate": 0.0003, "beta": 0.01, "epsilon": 0.2, "lambd": 0.95, "num_epoch": 3, "shared_critic": false, "learning_rate_schedule": "linear", "beta_schedule": "linear", "epsilon_schedule": "linear"}, "checkpoint_interval": 500000, "network_settings": {"normalize": false, "hidden_units": 512, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "reward_signals": {"extrinsic": {"gamma": 0.99, "strength": 1.0, "network_settings": {"normalize": false, "hidden_units": 128, "num_layers": 2, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}}, "rnd": {"gamma": 0.99, "strength": 0.01, "network_settings": {"normalize": false, "hidden_units": 64, "num_layers": 3, "vis_encode_type": "simple", "memory": null, "goal_conditioning_type": "hyper", "deterministic": false}, "learning_rate": 0.0001, "encoding_size": null}}, "init_path": null, "keep_checkpoints": 5, "even_checkpoints": false, "max_steps": 3000000, "time_horizon": 128, "summary_freq": 30000, "threaded": false, "self_play": null, "behavioral_cloning": null}}, "env_settings": {"env_path": "./training-envs-executables/linux/Pyramids/Pyramids", "env_args": null, "base_port": 5005, "num_envs": 1, "num_areas": 1, "seed": -1, "max_lifetime_restarts": 10, "restarts_rate_limit_n": 1, "restarts_rate_limit_period_s": 60}, "engine_settings": {"width": 84, "height": 84, "quality_level": 5, "time_scale": 20, "target_frame_rate": -1, "capture_frame_rate": 60, "no_graphics": true}, "environment_parameters": null, "checkpoint_settings": {"run_id": "Pyramids Training", "initialize_from": null, "load_model": false, "resume": false, "force": false, "train_model": false, "inference": false, "results_dir": "results"}, "torch_settings": {"device": null}, "debug": false}
|
configuration.yaml
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
default_settings: null
|
| 2 |
+
behaviors:
|
| 3 |
+
Pyramids:
|
| 4 |
+
trainer_type: ppo
|
| 5 |
+
hyperparameters:
|
| 6 |
+
batch_size: 128
|
| 7 |
+
buffer_size: 2048
|
| 8 |
+
learning_rate: 0.0003
|
| 9 |
+
beta: 0.01
|
| 10 |
+
epsilon: 0.2
|
| 11 |
+
lambd: 0.95
|
| 12 |
+
num_epoch: 3
|
| 13 |
+
shared_critic: false
|
| 14 |
+
learning_rate_schedule: linear
|
| 15 |
+
beta_schedule: linear
|
| 16 |
+
epsilon_schedule: linear
|
| 17 |
+
checkpoint_interval: 500000
|
| 18 |
+
network_settings:
|
| 19 |
+
normalize: false
|
| 20 |
+
hidden_units: 512
|
| 21 |
+
num_layers: 2
|
| 22 |
+
vis_encode_type: simple
|
| 23 |
+
memory: null
|
| 24 |
+
goal_conditioning_type: hyper
|
| 25 |
+
deterministic: false
|
| 26 |
+
reward_signals:
|
| 27 |
+
extrinsic:
|
| 28 |
+
gamma: 0.99
|
| 29 |
+
strength: 1.0
|
| 30 |
+
network_settings:
|
| 31 |
+
normalize: false
|
| 32 |
+
hidden_units: 128
|
| 33 |
+
num_layers: 2
|
| 34 |
+
vis_encode_type: simple
|
| 35 |
+
memory: null
|
| 36 |
+
goal_conditioning_type: hyper
|
| 37 |
+
deterministic: false
|
| 38 |
+
rnd:
|
| 39 |
+
gamma: 0.99
|
| 40 |
+
strength: 0.01
|
| 41 |
+
network_settings:
|
| 42 |
+
normalize: false
|
| 43 |
+
hidden_units: 64
|
| 44 |
+
num_layers: 3
|
| 45 |
+
vis_encode_type: simple
|
| 46 |
+
memory: null
|
| 47 |
+
goal_conditioning_type: hyper
|
| 48 |
+
deterministic: false
|
| 49 |
+
learning_rate: 0.0001
|
| 50 |
+
encoding_size: null
|
| 51 |
+
init_path: null
|
| 52 |
+
keep_checkpoints: 5
|
| 53 |
+
even_checkpoints: false
|
| 54 |
+
max_steps: 3000000
|
| 55 |
+
time_horizon: 128
|
| 56 |
+
summary_freq: 30000
|
| 57 |
+
threaded: false
|
| 58 |
+
self_play: null
|
| 59 |
+
behavioral_cloning: null
|
| 60 |
+
env_settings:
|
| 61 |
+
env_path: ./training-envs-executables/linux/Pyramids/Pyramids
|
| 62 |
+
env_args: null
|
| 63 |
+
base_port: 5005
|
| 64 |
+
num_envs: 1
|
| 65 |
+
num_areas: 1
|
| 66 |
+
seed: -1
|
| 67 |
+
max_lifetime_restarts: 10
|
| 68 |
+
restarts_rate_limit_n: 1
|
| 69 |
+
restarts_rate_limit_period_s: 60
|
| 70 |
+
engine_settings:
|
| 71 |
+
width: 84
|
| 72 |
+
height: 84
|
| 73 |
+
quality_level: 5
|
| 74 |
+
time_scale: 20
|
| 75 |
+
target_frame_rate: -1
|
| 76 |
+
capture_frame_rate: 60
|
| 77 |
+
no_graphics: true
|
| 78 |
+
environment_parameters: null
|
| 79 |
+
checkpoint_settings:
|
| 80 |
+
run_id: Pyramids Training
|
| 81 |
+
initialize_from: null
|
| 82 |
+
load_model: false
|
| 83 |
+
resume: false
|
| 84 |
+
force: false
|
| 85 |
+
train_model: false
|
| 86 |
+
inference: false
|
| 87 |
+
results_dir: results
|
| 88 |
+
torch_settings:
|
| 89 |
+
device: null
|
| 90 |
+
debug: false
|