Delete train_config.json

train_config.json

@@ -1,59 +0,0 @@
-{
-  "dataset_type": "short_metaworld",
-  "data_root": "data/short-metaworld-vla",
-  "train_jsonl": "",
-  "val_jsonl": "",
-  "val_ratio": 0.15,
-  "vision_model_name": "google/siglip2-base-patch16-224",
-  "text_model_name": "google/siglip2-base-patch16-224",
-  "separate_backbones": false,
-  "image_size": 224,
-  "freeze_vision": true,
-  "freeze_text": true,
-  "unfreeze_vision_last_n_layers": 2,
-  "unfreeze_text_last_n_layers": 2,
-  "fusion_type": "cross_attn",
-  "proj_dim": 512,
-  "fusion_hidden_dim": 1024,
-  "fusion_out_dim": 512,
-  "fusion_num_layers": 3,
-  "fusion_num_heads": 8,
-  "fusion_dropout": 0.1,
-  "normalize_embeddings": true,
-  "action_head_type": "moe",
-  "action_mlp_hidden_dim": 256,
-  "action_mlp_layers": 2,
-  "action_mlp_dropout": 0.1,
-  "moe_num_experts": 4,
-  "moe_hidden_dim": 512,
-  "moe_load_balance_weight": 0.001,
-  "router_condition": "action_input",
-  "act_chunk_size": 8,
-  "act_hidden_dim": 512,
-  "act_num_layers": 2,
-  "act_dropout": 0.1,
-  "use_geometry_features": true,
-  "geometry_dim": 6,
-  "temporal_context": 4,
-  "action_dim": 4,
-  "num_workers": 8,
-  "normalize_action_targets": true,
-  "action_norm_eps": 1e-06,
-  "learnable_action_scale": true,
-  "action_scale_init": 1.0,
-  "loss_type": "huber",
-  "huber_delta": 0.5,
-  "epochs": 80,
-  "batch_size": 32,
-  "grad_accum_steps": 2,
-  "learning_rate": 5e-05,
-  "weight_decay": 0.01,
-  "grad_clip_norm": 1.0,
-  "use_fp16": true,
-  "early_stopping_patience": 8,
-  "early_stopping_min_delta": 0.0005,
-  "seed": 42,
-  "device": "cuda",
-  "out_dir": "checkpoints_stage2_moe_full_peg_only",
-  "save_best_by_val": true
-}