forgedRice commited on
Commit
26f9d83
·
verified ·
1 Parent(s): 4b84cca

Pyramids model

Browse files
Files changed (4) hide show
  1. Pyramids.onnx +3 -0
  2. README.md +29 -0
  3. config.json +1 -0
  4. 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