{"pretrained_path": "hf://rail-berkeley/octo-base-1.5", "pretrained_step": null, "batch_size": 128, "shuffle_buffer_size": 10000, "num_steps": 1000000, "log_interval": 100, "eval_interval": 5000, "save_interval": 100000, "save_dir": "./checkpoints", "seed": 42, "wandb": {"project": "octo_finetune", "group": null, "entity": null}, "dataset_kwargs": {"name": "example_dataset", "data_dir": "/data/bangguo/maniskill_data", "image_obs_keys": {"primary": "image_primary"}, "proprio_obs_key": "proprio", "language_key": "language_instruction", "action_proprio_normalization_type": "normal", "action_normalization_mask": [true, true, true, true, true, true, false], "standardize_fn": {"module": "octo.data.oxe.oxe_standardization_transforms", "name": "bridge_dataset_transform", "args": [], "kwargs": {}}}, "modality": "multimodal", "finetuning_mode": "head_only", "window_size": 1, "optimizer": {"learning_rate": {"name": "cosine", "init_value": 0.0, "peak_value": 0.0003, "warmup_steps": 2000, "decay_steps": 1000000, "end_value": 0.0}, "weight_decay": 0.01, "clip_gradient": 1.0, "frozen_keys": ["octo_transformer.*"], "grad_accumulation_steps": null}, "val_kwargs": {"val_shuffle_buffer_size": 1000, "num_val_batches": 16}, "viz_kwargs": {"eval_batch_size": 128, "trajs_for_metrics": 100, "trajs_for_viz": 8, "samples_per_state": 8}, "frame_transform_threads": 16, "traj_transform_kwargs": {"window_size": 1, "action_horizon": 4, "goal_relabeling_strategy": "uniform", "task_augment_strategy": "delete_task_conditioning", "task_augment_kwargs": {"keep_image_prob": 0.5}}, "frame_transform_kwargs": {"resize_size": {"primary": [256, 256]}, "image_augment_kwargs": {"primary": {"random_resized_crop": {"scale": [0.8, 1.0], "ratio": [0.9, 1.1]}, "random_brightness": [0.1], "random_contrast": [0.9, 1.1], "random_saturation": [0.9, 1.1], "random_hue": [0.05], "augment_order": ["random_resized_crop", "random_brightness", "random_contrast", "random_saturation", "random_hue"]}, "wrist": {"random_brightness": [0.1], "random_contrast": [0.9, 1.1], "random_saturation": [0.9, 1.1], "random_hue": [0.05], "augment_order": ["random_brightness", "random_contrast", "random_saturation", "random_hue"]}}}}