File size: 1,278 Bytes
faacdc7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | {
"act_adaptor": "mlp3x_silu",
"act_pos_emb_config": [
[
"state",
1
],
[
"action",
16
]
],
"action_dim": 76,
"ema": {
"inv_gamma": 1.0,
"max_value": 0.9999,
"min_value": 0.0,
"power": 0.75,
"update_after_step": 0
},
"hrdt": {
"depth": 16,
"ffn_dim_multiplier": null,
"hidden_size": 2176,
"multiple_of": 256,
"norm_eps": 1e-05,
"num_heads": 16,
"num_kv_heads": 8,
"output_size": 76,
"use_flash_attn": true
},
"img_adapter": "mlp2x_silu",
"img_pos_emb_config": [
[
"image",
[
1,
1,
-729
]
]
],
"lang_adapter": "mlp2x_silu",
"lang_pos_emb_config": [
[
"language",
-1024
]
],
"loss_mode": null,
"max_img_len": 729,
"max_lang_len": 1024,
"mode": "pretrain",
"noise_scheduler": {
"num_inference_timesteps": 5,
"sampler_type": "uniform",
"timestep_max": 0.999
},
"pred_horizon": 16,
"pretrained_backbone_path": null,
"st_adaptor": "mlp3x_silu",
"state_dim": 76,
"text": {
"feature_dim": 4096
},
"time_noise": {
"a": 5,
"beta_m": 100
},
"training_mode": "lang",
"video_adapter": "mlp2x_silu",
"vision": {
"feature_dim": 2176
}
} |