{ "act_dim": 3, "action_tanh": true, "activation_function": "relu", "adv_act_dim": 3, "architectures": [ "AdversarialDT" ], "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": 1000 }