{ "batch_size": 192, "num_workers": 8, "shuffle_buffer_size": 1000, "normalization_mapping": { "ACTION": "ACTIONCHUNK_QUANTILE", "ENV": "QUANTILE", "STATE": "QUANTILE", "VISUAL": "IDENTITY" }, "features": { "action": { "type": "ACTION", "shape": "(20,)" }, "observation.image.0": { "type": "VISUAL", "shape": "(3, 224, 224)" }, "observation.image.1": { "type": "VISUAL", "shape": "(3, 224, 224)" }, "observation.image.2": { "type": "VISUAL", "shape": "(3, 224, 224)" }, "observation.state": { "type": "STATE", "shape": "(20,)" } }, "clip_values": { "ENV": [ -1.05, 1.05 ], "STATE": [ -1.05, 1.05 ] }, "chunk_size": null, "action_time_horizon_s": null, "min_action_chunk_size": 8, "observation_mapping": { "action.ee_6d_pos": "action", "observation.ee_6d_pos": "observation.state", "observation.images.agentview_image": "observation.image.0", "observation.images.left_wrist_image": "observation.image.1", "observation.images.right_wrist_image": "observation.image.2", "observation.language_instruction": "task" }, "observation_whitelist": null, "observation_blacklist": null, "transform_mapping": { "action": [ { "type": "convert_to_6d_actions", "action_key": "action", "action_type": "EE_6D_POS", "input_type": "Dict", "post_norm": false }, { "type": "delta_actions", "state_key": "observation.state", "action_key": "action", "action_type": "EE_6D_POS", "input_type": "Dict", "post_norm": false, "relative_to_state": true, "use_absolute_grippers": false } ], "observation.image.0": [ { "type": "resize_with_padding", "height": 224, "width": 224, "mode": "bilinear", "align_corners": false, "input_type": "Tensor" }, { "type": "color_jitter", "brightness": 0.2, "contrast": [ 0.8, 1.2 ], "saturation": [ 0.8, 1.2 ], "hue": 0.05, "input_type": "Tensor" } ], "observation.image.1": [ { "type": "resize_with_padding", "height": 224, "width": 224, "mode": "bilinear", "align_corners": false, "input_type": "Tensor" }, { "type": "color_jitter", "brightness": 0.2, "contrast": [ 0.8, 1.2 ], "saturation": [ 0.8, 1.2 ], "hue": 0.05, "input_type": "Tensor" } ], "observation.image.2": [ { "type": "resize_with_padding", "height": 224, "width": 224, "mode": "bilinear", "align_corners": false, "input_type": "Tensor" }, { "type": "color_jitter", "brightness": 0.2, "contrast": [ 0.8, 1.2 ], "saturation": [ 0.8, 1.2 ], "hue": 0.05, "input_type": "Tensor" } ], "observation.state": [ { "type": "convert_to_6d_actions", "action_key": "observation.state", "action_type": "EE_6D_POS", "input_type": "Dict", "post_norm": false } ] }, "language_instruction": null, "training_mode": "ROBOT_FLOWMATCH", "action_type": "EE_6D_POS", "serialize_stats": true, "video_decoder_cache_size": null, "repo_id": "tabletopsim_combined", "video_root_dir": null, "streaming": false, "tolerance_s": 0.1, "use_imagenet_stats": false, "split_name": "train", "prefetch_factor": 2, "max_episodes": null }