Upload new model run
Browse files- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/README.md +27 -0
- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/config.yaml +47 -0
- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/demo.gif +0 -0
- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/demo.mp4 +0 -0
- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/model.onnx +3 -0
- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/model.pt +3 -0
- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/model_config.yaml +23 -0
- models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/overrides.yaml +2 -0
- models/LunarLander-v2/Sac/results.yaml +1 -0
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/README.md
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
model-index:
|
| 3 |
+
- name: mock
|
| 4 |
+
results:
|
| 5 |
+
- task:
|
| 6 |
+
type: reinforcement-learning
|
| 7 |
+
name: reinforcement-learning
|
| 8 |
+
dataset:
|
| 9 |
+
name: mock
|
| 10 |
+
type: mock
|
| 11 |
+
metrics:
|
| 12 |
+
- type: mean_reward
|
| 13 |
+
value: mock +/- mock
|
| 14 |
+
name: mean_reward
|
| 15 |
+
verified: false
|
| 16 |
+
tags:
|
| 17 |
+
- reinforcement-learning
|
| 18 |
+
- connect-four
|
| 19 |
+
- pytorch
|
| 20 |
+
- custom-implementation
|
| 21 |
+
- generic
|
| 22 |
+
run_info:
|
| 23 |
+
litrl:
|
| 24 |
+
version: !!python/module:litrl._version ''
|
| 25 |
+
repo_url: https://github.com/c-gohlke/LitRL.git
|
| 26 |
+
commit_hash: 1fb3129526c84808b59ccb0bad7d6cf03c13c19f
|
| 27 |
+
---
|
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/config.yaml
ADDED
|
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
run_id: ${mlflow:null}
|
| 2 |
+
tags:
|
| 3 |
+
model: ${model._target_}
|
| 4 |
+
env: ${model.cfg.env_id}
|
| 5 |
+
run_id: ${run_id}
|
| 6 |
+
load_path: latest
|
| 7 |
+
model:
|
| 8 |
+
_target_: litrl.model.sac.model.Sac
|
| 9 |
+
cfg:
|
| 10 |
+
seed: 123
|
| 11 |
+
val_env_seed: 10000
|
| 12 |
+
env_id: LunarLander-v2
|
| 13 |
+
env_fabric:
|
| 14 |
+
_target_: litrl.make
|
| 15 |
+
id: ${model.cfg.env_id}
|
| 16 |
+
val_env_fabric:
|
| 17 |
+
_target_: litrl.make
|
| 18 |
+
id: ${model.cfg.env_id}
|
| 19 |
+
val: true
|
| 20 |
+
render_each_n_episodes: 1
|
| 21 |
+
lr: 0.0001
|
| 22 |
+
gamma: 0.99
|
| 23 |
+
warm_start_steps: 10000.0
|
| 24 |
+
hidden_size: 64
|
| 25 |
+
n_hidden_layers: 2
|
| 26 |
+
epsilon: 0
|
| 27 |
+
tau: 0.005
|
| 28 |
+
target_entropy: null
|
| 29 |
+
buffer:
|
| 30 |
+
batch_size: 256
|
| 31 |
+
max_size: 1000000.0
|
| 32 |
+
trainer:
|
| 33 |
+
_target_: lightning.Trainer
|
| 34 |
+
log_every_n_steps: 500
|
| 35 |
+
num_sanity_val_steps: 2
|
| 36 |
+
max_epochs: 500
|
| 37 |
+
limit_train_batches: 5000
|
| 38 |
+
logger:
|
| 39 |
+
_target_: litrl.common.logger.LitRLLogger
|
| 40 |
+
log_model: true
|
| 41 |
+
tags: ${tags}
|
| 42 |
+
run_id: ${run_id}
|
| 43 |
+
callbacks:
|
| 44 |
+
- _target_: litrl.common.huggingface.HuggingfaceCallback
|
| 45 |
+
mode: max
|
| 46 |
+
monitor: val_reward
|
| 47 |
+
filename: '{val_reward:.2f}-{step}'
|
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/demo.gif
ADDED
|
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/demo.mp4
ADDED
|
Binary file (21.7 kB). View file
|
|
|
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7c7b5ff7bcfd09d0fa37eaae9830af3599278eb7960eba053c4a5a7408886129
|
| 3 |
+
size 37631
|
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:75047aff371652523303e67fdcf902b3a7413255ce3993ee9b919879e78d9810
|
| 3 |
+
size 440074
|
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/model_config.yaml
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
root: !!python/object:litrl.schema.model.ModelConfigSchema
|
| 2 |
+
buffer: !!python/object:litrl.schema.model.BufferSchema
|
| 3 |
+
batch_size: 256
|
| 4 |
+
max_size: 1000000
|
| 5 |
+
env_fabric: !!python/object:litrl.schema.env.EnvSchema
|
| 6 |
+
_target_: litrl.make
|
| 7 |
+
id: LunarLander-v2
|
| 8 |
+
opponent: null
|
| 9 |
+
val: null
|
| 10 |
+
epsilon: 0.0
|
| 11 |
+
gamma: 0.99
|
| 12 |
+
hidden_size: 64
|
| 13 |
+
lr: 0.0001
|
| 14 |
+
n_hidden_layers: 2
|
| 15 |
+
seed: 123
|
| 16 |
+
target_entropy: null
|
| 17 |
+
tau: 0.005
|
| 18 |
+
val_env_fabric: !!python/object:litrl.schema.env.EnvSchema
|
| 19 |
+
_target_: litrl.make
|
| 20 |
+
id: LunarLander-v2
|
| 21 |
+
opponent: null
|
| 22 |
+
val: true
|
| 23 |
+
warm_start_steps: 10000
|
models/LunarLander-v2/Sac/c1150cc28abe4baa987213a6bfded48d/overrides.yaml
ADDED
|
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- model=sac
|
| 2 |
+
- env=lunar_lander
|
models/LunarLander-v2/Sac/results.yaml
CHANGED
|
@@ -5,6 +5,7 @@
|
|
| 5 |
6f8b3f77c2d84815986ac12b78860732: 21.669418334960938
|
| 6 |
796f8fe33b0344769264adafebc814a4: 984.3875732421875
|
| 7 |
901f3f0af1174ea5a8d81b607beac4b1: 74.01914978027344
|
|
|
|
| 8 |
c417006c10294a3e8c66bfabb74b5dcc: -85.7021713256836
|
| 9 |
dd63ef11719b45b1a3501ba0ae288223: 328.39361572265625
|
| 10 |
e0ff8028956548f7b27df0d3f142d154: 1777.4488525390625
|
|
|
|
| 5 |
6f8b3f77c2d84815986ac12b78860732: 21.669418334960938
|
| 6 |
796f8fe33b0344769264adafebc814a4: 984.3875732421875
|
| 7 |
901f3f0af1174ea5a8d81b607beac4b1: 74.01914978027344
|
| 8 |
+
c1150cc28abe4baa987213a6bfded48d: 138.86196899414062
|
| 9 |
c417006c10294a3e8c66bfabb74b5dcc: -85.7021713256836
|
| 10 |
dd63ef11719b45b1a3501ba0ae288223: 328.39361572265625
|
| 11 |
e0ff8028956548f7b27df0d3f142d154: 1777.4488525390625
|