FastWAM-LingBotVA-PurpleBox / lingbot-va /training_config.json
PencilHu's picture
Add files using upload-large-folder tool
1b0f2f6 verified
Raw
History Blame Contribute Delete
720 Bytes
{
"schema_version": "purple-box-lingbot-va-training-v1",
"checkpoint_step": 250,
"upstream": "Robbyant/lingbot-va",
"dataset": "jokeru/pick_right_purple_box_and_place_it_in_the_middle",
"train_episode_indices": [0, 1, 2, 3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 16, 17, 18, 19],
"held_out_episodes": [4, 15],
"world_size": 8,
"batch_size_per_gpu": 24,
"global_batch_size": 192,
"window_frames": 16,
"action_per_frame": 16,
"camera_keys": ["observation.images.right_eye", "observation.images.left_wrist", "observation.images.right_wrist"],
"source_action_indices": [7, 8, 9, 10, 11, 12, 13, 29],
"model_action_channel_ids": [21, 22, 23, 24, 25, 26, 27, 29],
"optimizer_state_included": false
}