{ | |
"act_dim": 3, | |
"action_tanh": true, | |
"activation_function": "relu", | |
"adv_act_dim": 3, | |
"architectures": [ | |
"VanillaDT" | |
], | |
"attn_pdrop": 0.1, | |
"bos_token_id": 50256, | |
"context_size": 20, | |
"embd_pdrop": 0.1, | |
"eos_token_id": 50256, | |
"flag": false, | |
"hidden_size": 128, | |
"initializer_range": 0.02, | |
"lambda1": 1.0, | |
"lambda2": 1.0, | |
"layer_norm_epsilon": 1e-05, | |
"log_interval_steps": 100, | |
"max_ep_len": 999, | |
"max_ep_return": 2720, | |
"max_obs_len": 999, | |
"max_obs_return": 2720, | |
"min_ep_return": 0.4963186, | |
"min_obs_return": 0.4963186, | |
"model_type": "decision_transformer", | |
"n_head": 1, | |
"n_inner": null, | |
"n_layer": 3, | |
"n_positions": 1024, | |
"pr_act_dim": 3, | |
"reorder_and_upcast_attn": false, | |
"resid_pdrop": 0.1, | |
"returns_scale": 1000, | |
"scale_attn_by_inverse_layer_idx": false, | |
"scale_attn_weights": true, | |
"state_dim": 11, | |
"state_mean": [ | |
1.2010676849198534, | |
0.049592377043302584, | |
-0.46424864468129656, | |
-0.132906587923632, | |
0.1598114346669523, | |
1.6783394392786875, | |
-0.24433229123756633, | |
-0.006172465230417297, | |
-0.5601292931238347, | |
-0.19438729466135996, | |
-0.5217454074382185 | |
], | |
"state_std": [ | |
0.16422606706562007, | |
0.07681961231530977, | |
0.34066130233739134, | |
0.22676278198501437, | |
0.5982900266887445, | |
0.9624198861913913, | |
1.1613482836051678, | |
0.7914457615241081, | |
1.675835772798825, | |
2.3415444950952553, | |
6.131794355269396 | |
], | |
"torch_dtype": "float32", | |
"total_train_steps": 10000, | |
"transformers_version": "4.29.2", | |
"use_cache": true, | |
"vocab_size": 1, | |
"warmup_steps": 1 | |
} | |