afonsosamarques's picture
Training in progress, step 3334
f96b9c5
raw
history blame
2 kB
{
"act_dim": 6,
"action_tanh": true,
"activation_function": "relu",
"adv_act_dim": 6,
"architectures": [
"MultipartARDT"
],
"attn_pdrop": 0.1,
"bos_token_id": 50256,
"context_size": 20,
"embd_pdrop": 0.1,
"eos_token_id": 50256,
"flag": 0,
"hidden_size": 128,
"initializer_range": 0.02,
"lambda1": 0.1,
"lambda2": 1.0,
"layer_norm_epsilon": 1e-05,
"log_interval_steps": 100,
"max_ep_len": 1000,
"max_ep_return": 2054.7955639899983,
"max_obs_len": 1000,
"max_obs_return": 2054.7955639899983,
"min_ep_return": -136.11629783300003,
"min_obs_return": -136.11629783300003,
"model_type": "decision_transformer",
"n_head": 1,
"n_inner": null,
"n_layer": 3,
"n_positions": 1024,
"pr_act_dim": 6,
"reorder_and_upcast_attn": false,
"resid_pdrop": 0.1,
"returns_scale": 1000,
"rtg_shift": 136.11639783300004,
"scale_attn_by_inverse_layer_idx": false,
"scale_attn_weights": true,
"state_dim": 17,
"state_mean": [
-0.11105085755165309,
0.03261898734092991,
0.09503744437931684,
-0.04665634126127169,
-0.1422736866683895,
0.11284505177282138,
-0.14719247154011844,
-0.20084285335399846,
1.6364819749735804,
-0.023428430703471302,
-0.011453850687947402,
0.028144657537117004,
-0.03049390117503992,
0.04205502023233107,
-0.047717887152182,
0.036133203776659344,
0.004414099071640206
],
"state_std": [
0.09256462065202084,
0.25102800281725657,
0.39216763873921784,
0.3014480322749965,
0.3594363079388843,
0.49460765306751897,
0.3719440989195892,
0.3381944556471006,
0.8053914288925733,
0.8879606428983204,
1.4583408884879427,
5.654339376346348,
4.917451449281009,
5.191545845751596,
5.222851276565701,
6.242544228826246,
5.14957441497535
],
"torch_dtype": "float32",
"total_train_steps": 10000,
"transformers_version": "4.31.0",
"use_cache": true,
"vocab_size": 1,
"warmup_steps": 1000
}