RoboSTAR / generator /config.json
Ivystream's picture
First
1d2cd4d
Raw History Blame Contribute Delete
673 Bytes
{
"body_codes": 96,
"hand_codes": 192,
"alpha_hand": 0.4,
"max_source_length": 512,
"max_motion_tokens": 100,
"use_compact_map": true,
"attn_implementation": "eager",
"stage_divisors": [
8,
4,
2,
1
],
"label_smoothing": 0.0,
"stage_loss_weights": [
1.25,
1.1,
1.0,
1.25
],
"hand_loss_weight": 1.25,
"length_loss_weight": 0.2,
"contrastive_loss_weight": 0.03,
"contrastive_temperature": 0.07,
"lm_family": "mt5",
"model_path": "google/mt5-large",
"token_namespace": "robotstar",
"scale_decoder_outputs": true,
"public_model_name": "RobotSTAR",
"format_version": "robotstar_generator_config_v1_1"
}