[
  {
    "label": "Stage1 baseline smoke",
    "group": "stage1",
    "phase": "baseline",
    "note": "Official-equivalent trainer, no OT router.",
    "rel_dir": "pact_finetune_route_32_20260516/stage1_smoke",
    "status": "ok",
    "steps": 1,
    "loss": 0.03042520396411419,
    "flow_mse": 0.03042520396411419,
    "art_l2": null,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {},
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_finetune_route_32_20260516/stage1_smoke",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 1e-06,
      "weight_decay": 0.01,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch"
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.03042520396411419,
        "flow_mse": 0.03042520396411419,
        "t_mean": 0.4452204704284668
      }
    ]
  },
  {
    "label": "OT first block",
    "group": "stage1",
    "phase": "phase1",
    "note": "One Stage1 cross-attention block patched with OT gating.",
    "rel_dir": "pact_stage1_ot_router_first_diag_smoke_20260520",
    "status": "ok",
    "steps": 1,
    "loss": 0.030395058915019035,
    "flow_mse": 0.030395058915019035,
    "art_l2": null,
    "ot_gate_mean": 0.98095703125,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "detach_plan": false
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_first_diag_smoke_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 1e-08,
      "optimizer": "adamw",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_detach_plan": false
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.030395058915019035,
        "flow_mse": 0.030395058915019035,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.98095703125,
        "ot_gate_min": 0.94921875,
        "ot_plan_mass": 1.0
      }
    ]
  },
  {
    "label": "OT mask prior",
    "group": "stage1",
    "phase": "phase2",
    "note": "Mask demoted from hard condition to OT cost prior.",
    "rel_dir": "pact_stage1_ot_router_maskprior_smoke_20260520",
    "status": "ok",
    "steps": 1,
    "loss": 0.02921455167233944,
    "flow_mse": 0.02921455167233944,
    "art_l2": null,
    "ot_gate_mean": 0.29248046875,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_maskprior_smoke_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 1e-08,
      "optimizer": "adamw",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.02921455167233944,
        "flow_mse": 0.02921455167233944,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29248046875,
        "ot_gate_min": 0.0208740234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      }
    ]
  },
  {
    "label": "OT mask prior 5-step",
    "group": "stage1",
    "phase": "phase2",
    "note": "SGD stability check for soft mask prior.",
    "rel_dir": "pact_stage1_ot_router_maskprior_5step_sgd1e8_20260520",
    "status": "ok",
    "steps": 5,
    "loss": 0.04972422122955322,
    "flow_mse": 0.04972422122955322,
    "art_l2": null,
    "ot_gate_mean": 0.2305908203125,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_maskprior_5step_sgd1e8_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 5,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.02921455167233944,
        "flow_mse": 0.02921455167233944,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29248046875,
        "ot_gate_min": 0.0208740234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 2,
        "loss": 0.06307478994131088,
        "flow_mse": 0.06307478994131088,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1976318359375,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 3,
        "loss": 0.060768574476242065,
        "flow_mse": 0.060768574476242065,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3642578125,
        "ot_gate_min": 0.022430419921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 4,
        "loss": 0.04302801936864853,
        "flow_mse": 0.04302801936864853,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26806640625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 5,
        "loss": 0.04972422122955322,
        "flow_mse": 0.04972422122955322,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2305908203125,
        "ot_gate_min": 0.0218353271484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      }
    ]
  },
  {
    "label": "OT RGB edge",
    "group": "stage1",
    "phase": "phase2",
    "note": "RGB Sobel edge regularized cost.",
    "rel_dir": "pact_stage1_ot_router_rgb_edge_1step_20260520",
    "status": "ok",
    "steps": 1,
    "loss": 0.02921142242848873,
    "flow_mse": 0.02921142242848873,
    "art_l2": null,
    "ot_gate_mean": 0.29248046875,
    "ot_edge_delta": 0.001398545573465526,
    "ot_virtual_mass": null,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_rgb_edge_1step_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 0.0,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.02921142242848873,
        "flow_mse": 0.02921142242848873,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001398545573465526,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29248046875,
        "ot_gate_min": 0.0208740234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      }
    ]
  },
  {
    "label": "OT edge 5-step",
    "group": "stage1",
    "phase": "phase2",
    "note": "Mask-edge regularized 5-step stability check.",
    "rel_dir": "pact_stage1_ot_router_edge_fixed_5step_20260520",
    "status": "ok",
    "steps": 5,
    "loss": 0.04973132163286209,
    "flow_mse": 0.04973132163286209,
    "art_l2": null,
    "ot_gate_mean": 0.23095703125,
    "ot_edge_delta": 0.0013192640617489815,
    "ot_virtual_mass": null,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_edge_fixed_5step_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 5,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.029213517904281616,
        "flow_mse": 0.029213517904281616,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015859350096434355,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29296875,
        "ot_gate_min": 0.0208740234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 2,
        "loss": 0.06307096779346466,
        "flow_mse": 0.06307096779346466,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012258578790351748,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1976318359375,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 3,
        "loss": 0.06076928228139877,
        "flow_mse": 0.06076928228139877,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011226125061511993,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3642578125,
        "ot_gate_min": 0.022430419921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 4,
        "loss": 0.04303176328539848,
        "flow_mse": 0.04303176328539848,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001303985365666449,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26806640625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      },
      {
        "step": 5,
        "loss": 0.04973132163286209,
        "flow_mse": 0.04973132163286209,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013192640617489815,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.23095703125,
        "ot_gate_min": 0.0218505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0
      }
    ]
  },
  {
    "label": "OT virtual patch",
    "group": "stage1",
    "phase": "phase3",
    "note": "Virtual image token with visible/virtual gate normalization.",
    "rel_dir": "pact_stage1_ot_router_virtual005_rgb_fixedgate_5step_20260520",
    "status": "ok",
    "steps": 5,
    "loss": 0.037792809307575226,
    "flow_mse": 0.037792809307575226,
    "art_l2": null,
    "ot_gate_mean": 0.2607421875,
    "ot_edge_delta": 0.0011901373509317636,
    "ot_virtual_mass": 0.04998779296875,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "virtual_tokens": 1,
      "virtual_mass": 0.05,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_virtual005_rgb_fixedgate_5step_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 5,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_virtual_tokens": 1,
      "ot_virtual_mass": 0.05,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.026793912053108215,
        "flow_mse": 0.026793912053108215,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009518907754682004,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.4013671875,
        "ot_gate_min": 0.02166748046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 2,
        "loss": 0.05393337458372116,
        "flow_mse": 0.05393337458372116,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010577188804745674,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.216064453125,
        "ot_gate_min": 0.0223236083984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 3,
        "loss": 0.05763113498687744,
        "flow_mse": 0.05763113498687744,
        "t_mean": 0.6866899728775024,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009937018621712923,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.156494140625,
        "ot_gate_min": 0.02197265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 4,
        "loss": 0.03923727944493294,
        "flow_mse": 0.03923727944493294,
        "t_mean": 0.5447205901145935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001060508075170219,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1578369140625,
        "ot_gate_min": 0.0216522216796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 5,
        "loss": 0.037792809307575226,
        "flow_mse": 0.037792809307575226,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011901373509317636,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2607421875,
        "ot_gate_min": 0.021331787109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      }
    ]
  },
  {
    "label": "OT first-third",
    "group": "stage1",
    "phase": "phase1",
    "note": "Patch first third of Stage1 cross-attention blocks.",
    "rel_dir": "pact_stage1_ot_router_firstthird_1step_20260520",
    "status": "ok",
    "steps": 1,
    "loss": 0.029234901070594788,
    "flow_mse": 0.029234901070594788,
    "art_l2": null,
    "ot_gate_mean": 0.290069580078125,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": 8.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first-third",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_firstthird_1step_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 0.0,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.029234901070594788,
        "flow_mse": 0.029234901070594788,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 8.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.290069580078125,
        "ot_gate_min": 0.020864486694335938,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0
      }
    ]
  },
  {
    "label": "OT edge first-third",
    "group": "stage1",
    "phase": "phase2",
    "note": "First-third patch plus edge-regularized cost.",
    "rel_dir": "pact_stage1_ot_router_edge_firstthird_fixed_1step_20260520",
    "status": "ok",
    "steps": 1,
    "loss": 0.029235010966658592,
    "flow_mse": 0.029235010966658592,
    "art_l2": null,
    "ot_gate_mean": 0.29052734375,
    "ot_edge_delta": 0.0015091553650563583,
    "ot_virtual_mass": null,
    "ot_modules_active": 8.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first-third",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage1_ot_router_edge_firstthird_fixed_1step_20260520",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 0.0,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.029235010966658592,
        "flow_mse": 0.029235010966658592,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0015091553650563583,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29052734375,
        "ot_gate_min": 0.020868301391601562,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0
      }
    ]
  },
  {
    "label": "Stage2 baseline smoke",
    "group": "stage2",
    "phase": "baseline",
    "note": "Official-equivalent Stage2 smoke.",
    "rel_dir": "pact_finetune_route_32_20260516/stage2_smoke",
    "status": "ok",
    "steps": 1,
    "loss": 3.2046496868133545,
    "flow_mse": 3.1097300052642822,
    "art_l2": 0.09491962939500809,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {},
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_finetune_route_32_20260516/stage2_smoke",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 1e-06,
      "weight_decay": 0.01,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch"
    },
    "logs": [
      {
        "step": 1,
        "loss": 3.2046496868133545,
        "flow_mse": 3.1097300052642822,
        "art_l2": 0.09491962939500809,
        "t_mean": 0.5347283482551575
      }
    ]
  },
  {
    "label": "Stage2 spatial diag",
    "group": "stage2",
    "phase": "phase4",
    "note": "Mask-derived spatial moment diagnostics for articulation rows.",
    "rel_dir": "pact_stage2_spatial_diag_smoke_20260521",
    "status": "ok",
    "steps": 1,
    "loss": 2.385582685470581,
    "flow_mse": 2.2749202251434326,
    "art_l2": 0.11066237837076187,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 5.0,
    "stage2_mask_empty_parts": 1.0,
    "stage2_mask_centroid_spread": 0.1345607340335846,
    "stage2_mask_spatial_var_mean": 0.03559853881597519,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {},
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage2_spatial_diag_smoke_20260521",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 0.0,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 2.385582685470581,
        "flow_mse": 2.2749202251434326,
        "art_l2": 0.11066237837076187,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 5.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.097297303378582,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1345607340335846,
        "stage2_mask_spatial_var_mean": 0.03559853881597519
      }
    ]
  },
  {
    "label": "Stage2 spatial adapter 1-step",
    "group": "stage2",
    "phase": "phase4",
    "note": "Zero-initialized articulation residual adapter smoke.",
    "rel_dir": "pact_stage2_spatial_adapter_smoke_20260521",
    "status": "ok",
    "steps": 1,
    "loss": 7.411160945892334,
    "flow_mse": 7.277588367462158,
    "art_l2": 0.13357242941856384,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 3.0,
    "stage2_mask_empty_parts": 1.0,
    "stage2_mask_centroid_spread": 0.16794748604297638,
    "stage2_mask_spatial_var_mean": 0.04503370076417923,
    "stage2_spatial_adapter_l2": 0.0,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": true,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage2_spatial_adapter_smoke_20260521",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 1,
      "lr": 0.0,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": true,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 7.411160945892334,
        "flow_mse": 7.277588367462158,
        "art_l2": 0.13357242941856384,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14974434673786163,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16794748604297638,
        "stage2_mask_spatial_var_mean": 0.04503370076417923,
        "stage2_spatial_adapter_l2": 0.0
      }
    ]
  },
  {
    "label": "Stage2 spatial adapter 5-step",
    "group": "stage2",
    "phase": "phase4",
    "note": "Spatial adapter 5-step SGD stability check.",
    "rel_dir": "pact_stage2_spatial_adapter_5step_20260521",
    "status": "ok",
    "steps": 5,
    "loss": 3.3669536113739014,
    "flow_mse": 3.2528350353240967,
    "art_l2": 0.11411850899457932,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 6.0,
    "stage2_mask_empty_parts": 2.0,
    "stage2_mask_centroid_spread": 0.12281860411167145,
    "stage2_mask_spatial_var_mean": 0.0692753717303276,
    "stage2_spatial_adapter_l2": 1.7222661193919904e-22,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": true,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_stage2_spatial_adapter_5step_20260521",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 5,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": true,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 7.4107770919799805,
        "flow_mse": 7.277140140533447,
        "art_l2": 0.13363701105117798,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14974434673786163,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16794748604297638,
        "stage2_mask_spatial_var_mean": 0.04503370076417923,
        "stage2_spatial_adapter_l2": 0.0
      },
      {
        "step": 2,
        "loss": 3.420111656188965,
        "flow_mse": 3.3164515495300293,
        "art_l2": 0.10366011410951614,
        "t_mean": 0.3323965072631836,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.1470051258802414,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.21312789618968964,
        "stage2_mask_spatial_var_mean": 0.052936047315597534,
        "stage2_spatial_adapter_l2": 3.0293339499927006e-24
      },
      {
        "step": 3,
        "loss": 4.469086170196533,
        "flow_mse": 4.312153339385986,
        "art_l2": 0.15693266689777374,
        "t_mean": 0.8121371269226074,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09739469736814499,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15333124995231628,
        "stage2_mask_spatial_var_mean": 0.054026901721954346,
        "stage2_spatial_adapter_l2": 1.5986886288975308e-23
      },
      {
        "step": 4,
        "loss": 7.818053245544434,
        "flow_mse": 7.688683986663818,
        "art_l2": 0.12936925888061523,
        "t_mean": 0.686140239238739,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14919649064540863,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17030486464500427,
        "stage2_mask_spatial_var_mean": 0.0551006905734539,
        "stage2_spatial_adapter_l2": 1.303023782501749e-22
      },
      {
        "step": 5,
        "loss": 3.3669536113739014,
        "flow_mse": 3.2528350353240967,
        "art_l2": 0.11411850899457932,
        "t_mean": 0.728419840335846,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08144631236791611,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12281860411167145,
        "stage2_mask_spatial_var_mean": 0.0692753717303276,
        "stage2_spatial_adapter_l2": 1.7222661193919904e-22
      }
    ]
  },
  {
    "label": "Matrix Stage1 baseline",
    "group": "stage1",
    "phase": "matrix",
    "note": "Quick matrix baseline with the official-equivalent Stage1 path.",
    "rel_dir": "pact_ot_matrix_20260521_stage1_baseline",
    "status": "ok",
    "steps": 8,
    "loss": 1.0630731582641602,
    "flow_mse": 1.0630731582641602,
    "art_l2": null,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage1_baseline",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 8,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169
      }
    ]
  },
  {
    "label": "Matrix Stage1 OT first",
    "group": "stage1",
    "phase": "matrix",
    "note": "Quick matrix with one OT-gated cross-attention block.",
    "rel_dir": "pact_ot_matrix_20260521_stage1_ot_first",
    "status": "ok",
    "steps": 8,
    "loss": 1.0630731582641602,
    "flow_mse": 1.0630731582641602,
    "art_l2": null,
    "ot_gate_mean": 0.97509765625,
    "ot_edge_delta": 0.0,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage1_ot_first",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 8,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.970703125,
        "ot_gate_min": 0.935546875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96923828125,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97119140625,
        "ot_gate_min": 0.943359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9736328125,
        "ot_gate_min": 0.943359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97119140625,
        "ot_gate_min": 0.94091796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9697265625,
        "ot_gate_min": 0.93212890625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9677734375,
        "ot_gate_min": 0.92626953125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97509765625,
        "ot_gate_min": 0.95166015625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Matrix Stage1 mask prior",
    "group": "stage1",
    "phase": "matrix",
    "note": "Quick matrix with mask labels as a soft OT prior.",
    "rel_dir": "pact_ot_matrix_20260521_stage1_mask_prior",
    "status": "ok",
    "steps": 8,
    "loss": 1.0630731582641602,
    "flow_mse": 1.0630731582641602,
    "art_l2": null,
    "ot_gate_mean": 0.32373046875,
    "ot_edge_delta": 0.0,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage1_mask_prior",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 8,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2919921875,
        "ot_gate_min": 0.0208587646484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.19677734375,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.36572265625,
        "ot_gate_min": 0.0225067138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.267822265625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.230224609375,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2335205078125,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.336181640625,
        "ot_gate_min": 0.0218505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32373046875,
        "ot_gate_min": 0.02618408203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Matrix Stage1 RGB edge",
    "group": "stage1",
    "phase": "matrix",
    "note": "Quick matrix with RGB Sobel edge-regularized OT cost.",
    "rel_dir": "pact_ot_matrix_20260521_stage1_rgb_edge",
    "status": "ok",
    "steps": 8,
    "loss": 1.0630731582641602,
    "flow_mse": 1.0630731582641602,
    "art_l2": null,
    "ot_gate_mean": 0.32373046875,
    "ot_edge_delta": 0.001582509372383356,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage1_rgb_edge",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 8,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012463603634387255,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2919921875,
        "ot_gate_min": 0.0208587646484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013847764348611236,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1971435546875,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001398036372847855,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3662109375,
        "ot_gate_min": 0.02252197265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001753590302541852,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.267822265625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013486214447766542,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.23046875,
        "ot_gate_min": 0.02178955078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001678119762800634,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2340087890625,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014689782401546836,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.336669921875,
        "ot_gate_min": 0.0218505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001582509372383356,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32373046875,
        "ot_gate_min": 0.0261993408203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Matrix Stage1 virtual patch",
    "group": "stage1",
    "phase": "matrix",
    "note": "Quick matrix with a virtual patch channel.",
    "rel_dir": "pact_ot_matrix_20260521_stage1_virtual_patch",
    "status": "ok",
    "steps": 8,
    "loss": 1.033375859260559,
    "flow_mse": 1.033375859260559,
    "art_l2": null,
    "ot_gate_mean": 0.146240234375,
    "ot_edge_delta": 0.0014481099788099527,
    "ot_virtual_mass": 0.04998779296875,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "virtual_tokens": 1,
      "virtual_mass": 0.05,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage1_virtual_patch",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 8,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_virtual_tokens": 1,
      "ot_virtual_mass": 0.05,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0175116062164307,
        "flow_mse": 1.0175116062164307,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013981556985527277,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.400634765625,
        "ot_gate_min": 0.0216217041015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 2,
        "loss": 1.0188701152801514,
        "flow_mse": 1.0188701152801514,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013099866919219494,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2154541015625,
        "ot_gate_min": 0.022247314453125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 3,
        "loss": 1.0372271537780762,
        "flow_mse": 1.0372271537780762,
        "t_mean": 0.6866899728775024,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013684009900316596,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.15869140625,
        "ot_gate_min": 0.0219573974609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 4,
        "loss": 1.0259208679199219,
        "flow_mse": 1.0259208679199219,
        "t_mean": 0.5447205901145935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001627056859433651,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.158203125,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 5,
        "loss": 1.0232212543487549,
        "flow_mse": 1.0232212543487549,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015082642203196883,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.260986328125,
        "ot_gate_min": 0.0213623046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 6,
        "loss": 1.0559325218200684,
        "flow_mse": 1.0559325218200684,
        "t_mean": 0.681830883026123,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012014677049592137,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29248046875,
        "ot_gate_min": 0.0252532958984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 7,
        "loss": 1.0603113174438477,
        "flow_mse": 1.0603113174438477,
        "t_mean": 0.7412480115890503,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014335500309243798,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.349365234375,
        "ot_gate_min": 0.0259857177734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 8,
        "loss": 1.033375859260559,
        "flow_mse": 1.033375859260559,
        "t_mean": 0.39051905274391174,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014481099788099527,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.146240234375,
        "ot_gate_min": 0.021942138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      }
    ]
  },
  {
    "label": "Matrix Stage1 first-third edge",
    "group": "stage1",
    "phase": "matrix",
    "note": "Quick matrix with OT injected into the first third of Stage1 blocks.",
    "rel_dir": "pact_ot_matrix_20260521_stage1_firstthird_edge",
    "status": "ok",
    "steps": 3,
    "loss": 1.0302751064300537,
    "flow_mse": 1.0302751064300537,
    "art_l2": null,
    "ot_gate_mean": 0.36029052734375,
    "ot_edge_delta": 0.0018975692219100893,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 8.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first-third",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage1_firstthird_edge",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 3,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001615208646398969,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.280731201171875,
        "ot_gate_min": 0.020847320556640625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001815385592635721,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.195343017578125,
        "ot_gate_min": 0.022920608520507812,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0018975692219100893,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.36029052734375,
        "ot_gate_min": 0.022266387939453125,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Matrix Stage2 spatial diag",
    "group": "stage2",
    "phase": "matrix",
    "note": "Quick matrix Stage2 spatial diagnostics.",
    "rel_dir": "pact_ot_matrix_20260521_stage2_spatial_diag",
    "status": "ok",
    "steps": 5,
    "loss": 3.4941177368164062,
    "flow_mse": 3.2936394214630127,
    "art_l2": 0.2004782110452652,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 4.0,
    "stage2_mask_empty_parts": 2.0,
    "stage2_mask_centroid_spread": 0.12117581814527512,
    "stage2_mask_spatial_var_mean": 0.06845814734697342,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage2_spatial_diag",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 5,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 2.384955406188965,
        "flow_mse": 2.2743406295776367,
        "art_l2": 0.11061479896306992,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 5.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.097297303378582,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1345607340335846,
        "stage2_mask_spatial_var_mean": 0.03559853881597519
      },
      {
        "step": 2,
        "loss": 1.0193581581115723,
        "flow_mse": 0.9344261884689331,
        "art_l2": 0.08493191003799438,
        "t_mean": 0.3323965072631836,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.07617656141519547,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11268322169780731,
        "stage2_mask_spatial_var_mean": 0.024656636640429497
      },
      {
        "step": 3,
        "loss": 4.097375869750977,
        "flow_mse": 4.015601634979248,
        "art_l2": 0.08177444338798523,
        "t_mean": 0.8121371269226074,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.1260043829679489,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13602715730667114,
        "stage2_mask_spatial_var_mean": 0.055630188435316086
      },
      {
        "step": 4,
        "loss": 2.3762500286102295,
        "flow_mse": 2.3050081729888916,
        "art_l2": 0.07124177366495132,
        "t_mean": 0.686140239238739,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09447285532951355,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12284307181835175,
        "stage2_mask_spatial_var_mean": 0.03497007489204407
      },
      {
        "step": 5,
        "loss": 3.4941177368164062,
        "flow_mse": 3.2936394214630127,
        "art_l2": 0.2004782110452652,
        "t_mean": 0.728419840335846,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.13586559891700745,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12117581814527512,
        "stage2_mask_spatial_var_mean": 0.06845814734697342
      }
    ]
  },
  {
    "label": "Matrix Stage2 spatial adapter",
    "group": "stage2",
    "phase": "matrix",
    "note": "Quick matrix Stage2 spatial adapter.",
    "rel_dir": "pact_ot_matrix_20260521_stage2_spatial_adapter",
    "status": "ok",
    "steps": 5,
    "loss": 3.367434024810791,
    "flow_mse": 3.2533464431762695,
    "art_l2": 0.11408749967813492,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 6.0,
    "stage2_mask_empty_parts": 2.0,
    "stage2_mask_centroid_spread": 0.12281860411167145,
    "stage2_mask_spatial_var_mean": 0.0692753717303276,
    "stage2_spatial_adapter_l2": 1.7224368920007526e-22,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": true,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_stage2_spatial_adapter",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 5,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": true,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": false,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 7.409876823425293,
        "flow_mse": 7.27638053894043,
        "art_l2": 0.13349612057209015,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14974434673786163,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16794748604297638,
        "stage2_mask_spatial_var_mean": 0.04503370076417923,
        "stage2_spatial_adapter_l2": 0.0
      },
      {
        "step": 2,
        "loss": 3.420111656188965,
        "flow_mse": 3.3164515495300293,
        "art_l2": 0.10366011410951614,
        "t_mean": 0.3323965072631836,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.1470051258802414,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.21312789618968964,
        "stage2_mask_spatial_var_mean": 0.052936047315597534,
        "stage2_spatial_adapter_l2": 3.0245786964560284e-24
      },
      {
        "step": 3,
        "loss": 4.469086170196533,
        "flow_mse": 4.312153339385986,
        "art_l2": 0.15693266689777374,
        "t_mean": 0.8121371269226074,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09739469736814499,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15333124995231628,
        "stage2_mask_spatial_var_mean": 0.054026901721954346,
        "stage2_spatial_adapter_l2": 1.5988661226012055e-23
      },
      {
        "step": 4,
        "loss": 7.818053245544434,
        "flow_mse": 7.688683986663818,
        "art_l2": 0.12936925888061523,
        "t_mean": 0.686140239238739,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14919649064540863,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17030486464500427,
        "stage2_mask_spatial_var_mean": 0.0551006905734539,
        "stage2_spatial_adapter_l2": 1.3031613598436196e-22
      },
      {
        "step": 5,
        "loss": 3.367434024810791,
        "flow_mse": 3.2533464431762695,
        "art_l2": 0.11408749967813492,
        "t_mean": 0.728419840335846,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08144631236791611,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12281860411167145,
        "stage2_mask_spatial_var_mean": 0.0692753717303276,
        "stage2_spatial_adapter_l2": 1.7224368920007526e-22
      }
    ]
  },
  {
    "label": "Full Stage1 baseline",
    "group": "stage1",
    "phase": "full",
    "note": "Full matrix baseline with final checkpoint export.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage1_baseline",
    "status": "ok",
    "steps": 60,
    "loss": 1.0311006307601929,
    "flow_mse": 1.0311006307601929,
    "art_l2": null,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage1_baseline",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169
      },
      {
        "step": 9,
        "loss": 1.0248684883117676,
        "flow_mse": 1.0248684883117676,
        "t_mean": 0.5298079252243042
      },
      {
        "step": 10,
        "loss": 1.0194365978240967,
        "flow_mse": 1.0194365978240967,
        "t_mean": 0.5381855368614197
      },
      {
        "step": 11,
        "loss": 1.059285044670105,
        "flow_mse": 1.059285044670105,
        "t_mean": 0.5974305868148804
      },
      {
        "step": 12,
        "loss": 1.030677318572998,
        "flow_mse": 1.030677318572998,
        "t_mean": 0.4660314917564392
      },
      {
        "step": 13,
        "loss": 1.0135877132415771,
        "flow_mse": 1.0135877132415771,
        "t_mean": 0.42932528257369995
      },
      {
        "step": 14,
        "loss": 1.0318082571029663,
        "flow_mse": 1.0318082571029663,
        "t_mean": 0.44856566190719604
      },
      {
        "step": 15,
        "loss": 1.027304768562317,
        "flow_mse": 1.027304768562317,
        "t_mean": 0.34305432438850403
      },
      {
        "step": 16,
        "loss": 1.0195131301879883,
        "flow_mse": 1.0195131301879883,
        "t_mean": 0.4998842179775238
      },
      {
        "step": 17,
        "loss": 1.0307263135910034,
        "flow_mse": 1.0307263135910034,
        "t_mean": 0.4249878525733948
      },
      {
        "step": 18,
        "loss": 1.0178287029266357,
        "flow_mse": 1.0178287029266357,
        "t_mean": 0.42435136437416077
      },
      {
        "step": 19,
        "loss": 1.0279525518417358,
        "flow_mse": 1.0279525518417358,
        "t_mean": 0.49784499406814575
      },
      {
        "step": 20,
        "loss": 1.03252112865448,
        "flow_mse": 1.03252112865448,
        "t_mean": 0.4546642005443573
      },
      {
        "step": 21,
        "loss": 1.0540964603424072,
        "flow_mse": 1.0540964603424072,
        "t_mean": 0.31838786602020264
      },
      {
        "step": 22,
        "loss": 1.0650438070297241,
        "flow_mse": 1.0650438070297241,
        "t_mean": 0.5307409167289734
      },
      {
        "step": 23,
        "loss": 1.0473723411560059,
        "flow_mse": 1.0473723411560059,
        "t_mean": 0.511449933052063
      },
      {
        "step": 24,
        "loss": 1.0280429124832153,
        "flow_mse": 1.0280429124832153,
        "t_mean": 0.5524937510490417
      },
      {
        "step": 25,
        "loss": 1.0299099683761597,
        "flow_mse": 1.0299099683761597,
        "t_mean": 0.4828733801841736
      },
      {
        "step": 26,
        "loss": 1.0370187759399414,
        "flow_mse": 1.0370187759399414,
        "t_mean": 0.47976693511009216
      },
      {
        "step": 27,
        "loss": 1.0062942504882812,
        "flow_mse": 1.0062942504882812,
        "t_mean": 0.7675442695617676
      },
      {
        "step": 28,
        "loss": 1.099644660949707,
        "flow_mse": 1.099644660949707,
        "t_mean": 0.6012675166130066
      },
      {
        "step": 29,
        "loss": 1.0200495719909668,
        "flow_mse": 1.0200495719909668,
        "t_mean": 0.6016367673873901
      },
      {
        "step": 30,
        "loss": 1.0361602306365967,
        "flow_mse": 1.0361602306365967,
        "t_mean": 0.5350275039672852
      },
      {
        "step": 31,
        "loss": 1.023235559463501,
        "flow_mse": 1.023235559463501,
        "t_mean": 0.3996628522872925
      },
      {
        "step": 32,
        "loss": 1.0194340944290161,
        "flow_mse": 1.0194340944290161,
        "t_mean": 0.4242017865180969
      },
      {
        "step": 33,
        "loss": 1.0286738872528076,
        "flow_mse": 1.0286738872528076,
        "t_mean": 0.6198998689651489
      },
      {
        "step": 34,
        "loss": 1.0266057252883911,
        "flow_mse": 1.0266057252883911,
        "t_mean": 0.4256276786327362
      },
      {
        "step": 35,
        "loss": 1.021877646446228,
        "flow_mse": 1.021877646446228,
        "t_mean": 0.5468468070030212
      },
      {
        "step": 36,
        "loss": 1.035739779472351,
        "flow_mse": 1.035739779472351,
        "t_mean": 0.4543689489364624
      },
      {
        "step": 37,
        "loss": 1.0911012887954712,
        "flow_mse": 1.0911012887954712,
        "t_mean": 0.5334687829017639
      },
      {
        "step": 38,
        "loss": 1.0154587030410767,
        "flow_mse": 1.0154587030410767,
        "t_mean": 0.4061962068080902
      },
      {
        "step": 39,
        "loss": 1.030579924583435,
        "flow_mse": 1.030579924583435,
        "t_mean": 0.6304954886436462
      },
      {
        "step": 40,
        "loss": 1.041369080543518,
        "flow_mse": 1.041369080543518,
        "t_mean": 0.4235572814941406
      },
      {
        "step": 41,
        "loss": 1.0356323719024658,
        "flow_mse": 1.0356323719024658,
        "t_mean": 0.3948693871498108
      },
      {
        "step": 42,
        "loss": 1.1201456785202026,
        "flow_mse": 1.1201456785202026,
        "t_mean": 0.5838073492050171
      },
      {
        "step": 43,
        "loss": 1.039523959159851,
        "flow_mse": 1.039523959159851,
        "t_mean": 0.674645185470581
      },
      {
        "step": 44,
        "loss": 1.0210646390914917,
        "flow_mse": 1.0210646390914917,
        "t_mean": 0.4860682487487793
      },
      {
        "step": 45,
        "loss": 1.0069496631622314,
        "flow_mse": 1.0069496631622314,
        "t_mean": 0.3828374445438385
      },
      {
        "step": 46,
        "loss": 1.0390965938568115,
        "flow_mse": 1.0390965938568115,
        "t_mean": 0.3058706820011139
      },
      {
        "step": 47,
        "loss": 1.0197845697402954,
        "flow_mse": 1.0197845697402954,
        "t_mean": 0.31310853362083435
      },
      {
        "step": 48,
        "loss": 1.0371644496917725,
        "flow_mse": 1.0371644496917725,
        "t_mean": 0.5626304149627686
      },
      {
        "step": 49,
        "loss": 1.036482810974121,
        "flow_mse": 1.036482810974121,
        "t_mean": 0.7366573214530945
      },
      {
        "step": 50,
        "loss": 1.0249230861663818,
        "flow_mse": 1.0249230861663818,
        "t_mean": 0.5806166529655457
      },
      {
        "step": 51,
        "loss": 1.0295422077178955,
        "flow_mse": 1.0295422077178955,
        "t_mean": 0.6815006732940674
      },
      {
        "step": 52,
        "loss": 1.0306437015533447,
        "flow_mse": 1.0306437015533447,
        "t_mean": 0.509903609752655
      },
      {
        "step": 53,
        "loss": 1.016850233078003,
        "flow_mse": 1.016850233078003,
        "t_mean": 0.5933268070220947
      },
      {
        "step": 54,
        "loss": 1.0910499095916748,
        "flow_mse": 1.0910499095916748,
        "t_mean": 0.5004596710205078
      },
      {
        "step": 55,
        "loss": 1.052876353263855,
        "flow_mse": 1.052876353263855,
        "t_mean": 0.7171943187713623
      },
      {
        "step": 56,
        "loss": 1.0530800819396973,
        "flow_mse": 1.0530800819396973,
        "t_mean": 0.4547177255153656
      },
      {
        "step": 57,
        "loss": 1.0307668447494507,
        "flow_mse": 1.0307668447494507,
        "t_mean": 0.43234071135520935
      },
      {
        "step": 58,
        "loss": 1.0234510898590088,
        "flow_mse": 1.0234510898590088,
        "t_mean": 0.5147815942764282
      },
      {
        "step": 59,
        "loss": 1.0753936767578125,
        "flow_mse": 1.0753936767578125,
        "t_mean": 0.49480095505714417
      },
      {
        "step": 60,
        "loss": 1.0311006307601929,
        "flow_mse": 1.0311006307601929,
        "t_mean": 0.4051949977874756
      }
    ]
  },
  {
    "label": "Full Stage1 OT first",
    "group": "stage1",
    "phase": "full",
    "note": "Full matrix with one OT-gated cross-attention block.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage1_ot_first",
    "status": "ok",
    "steps": 60,
    "loss": 1.0311006307601929,
    "flow_mse": 1.0311006307601929,
    "art_l2": null,
    "ot_gate_mean": 0.970703125,
    "ot_edge_delta": 0.0,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage1_ot_first",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.970703125,
        "ot_gate_min": 0.935546875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96923828125,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97119140625,
        "ot_gate_min": 0.943359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9736328125,
        "ot_gate_min": 0.943359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97119140625,
        "ot_gate_min": 0.94091796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9697265625,
        "ot_gate_min": 0.93212890625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9677734375,
        "ot_gate_min": 0.92626953125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97509765625,
        "ot_gate_min": 0.95166015625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 1.0248684883117676,
        "flow_mse": 1.0248684883117676,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9677734375,
        "ot_gate_min": 0.9296875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 1.0194365978240967,
        "flow_mse": 1.0194365978240967,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.974609375,
        "ot_gate_min": 0.943359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 1.059285044670105,
        "flow_mse": 1.059285044670105,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97314453125,
        "ot_gate_min": 0.94580078125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 1.030677318572998,
        "flow_mse": 1.030677318572998,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9716796875,
        "ot_gate_min": 0.93896484375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 1.0135877132415771,
        "flow_mse": 1.0135877132415771,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97509765625,
        "ot_gate_min": 0.95068359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 1.0318082571029663,
        "flow_mse": 1.0318082571029663,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9736328125,
        "ot_gate_min": 0.94580078125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 1.027304768562317,
        "flow_mse": 1.027304768562317,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.966796875,
        "ot_gate_min": 0.93212890625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 1.0195131301879883,
        "flow_mse": 1.0195131301879883,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.970703125,
        "ot_gate_min": 0.935546875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 1.0307263135910034,
        "flow_mse": 1.0307263135910034,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.974609375,
        "ot_gate_min": 0.95068359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 1.0178287029266357,
        "flow_mse": 1.0178287029266357,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96630859375,
        "ot_gate_min": 0.93115234375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 1.0279525518417358,
        "flow_mse": 1.0279525518417358,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96630859375,
        "ot_gate_min": 0.9287109375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 1.03252112865448,
        "flow_mse": 1.03252112865448,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9697265625,
        "ot_gate_min": 0.93798828125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 21,
        "loss": 1.0540964603424072,
        "flow_mse": 1.0540964603424072,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97607421875,
        "ot_gate_min": 0.94580078125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 22,
        "loss": 1.0650438070297241,
        "flow_mse": 1.0650438070297241,
        "t_mean": 0.5307409167289734,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9736328125,
        "ot_gate_min": 0.9482421875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 23,
        "loss": 1.0473723411560059,
        "flow_mse": 1.0473723411560059,
        "t_mean": 0.511449933052063,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97509765625,
        "ot_gate_min": 0.95166015625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 24,
        "loss": 1.0280429124832153,
        "flow_mse": 1.0280429124832153,
        "t_mean": 0.5524937510490417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9658203125,
        "ot_gate_min": 0.93212890625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 25,
        "loss": 1.0299099683761597,
        "flow_mse": 1.0299099683761597,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97314453125,
        "ot_gate_min": 0.9423828125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 26,
        "loss": 1.0370187759399414,
        "flow_mse": 1.0370187759399414,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97265625,
        "ot_gate_min": 0.94091796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 27,
        "loss": 1.0062942504882812,
        "flow_mse": 1.0062942504882812,
        "t_mean": 0.7675442695617676,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97998046875,
        "ot_gate_min": 0.958984375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 28,
        "loss": 1.099644660949707,
        "flow_mse": 1.099644660949707,
        "t_mean": 0.6012675166130066,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.970703125,
        "ot_gate_min": 0.927734375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 29,
        "loss": 1.0200495719909668,
        "flow_mse": 1.0200495719909668,
        "t_mean": 0.6016367673873901,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97119140625,
        "ot_gate_min": 0.93212890625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 30,
        "loss": 1.0361602306365967,
        "flow_mse": 1.0361602306365967,
        "t_mean": 0.5350275039672852,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.970703125,
        "ot_gate_min": 0.93701171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 31,
        "loss": 1.023235559463501,
        "flow_mse": 1.023235559463501,
        "t_mean": 0.3996628522872925,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97119140625,
        "ot_gate_min": 0.93701171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 32,
        "loss": 1.0194340944290161,
        "flow_mse": 1.0194340944290161,
        "t_mean": 0.4242017865180969,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96533203125,
        "ot_gate_min": 0.93115234375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 33,
        "loss": 1.0286738872528076,
        "flow_mse": 1.0286738872528076,
        "t_mean": 0.6198998689651489,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.966796875,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 34,
        "loss": 1.0266057252883911,
        "flow_mse": 1.0266057252883911,
        "t_mean": 0.4256276786327362,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97314453125,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 35,
        "loss": 1.021877646446228,
        "flow_mse": 1.021877646446228,
        "t_mean": 0.5468468070030212,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96044921875,
        "ot_gate_min": 0.91064453125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 36,
        "loss": 1.035739779472351,
        "flow_mse": 1.035739779472351,
        "t_mean": 0.4543689489364624,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9677734375,
        "ot_gate_min": 0.9287109375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 37,
        "loss": 1.0911012887954712,
        "flow_mse": 1.0911012887954712,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9677734375,
        "ot_gate_min": 0.9287109375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 38,
        "loss": 1.0154587030410767,
        "flow_mse": 1.0154587030410767,
        "t_mean": 0.4061962068080902,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96533203125,
        "ot_gate_min": 0.923828125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 39,
        "loss": 1.030579924583435,
        "flow_mse": 1.030579924583435,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97021484375,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 40,
        "loss": 1.041369080543518,
        "flow_mse": 1.041369080543518,
        "t_mean": 0.4235572814941406,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96923828125,
        "ot_gate_min": 0.93896484375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 41,
        "loss": 1.0356323719024658,
        "flow_mse": 1.0356323719024658,
        "t_mean": 0.3948693871498108,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9775390625,
        "ot_gate_min": 0.95654296875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 42,
        "loss": 1.1201456785202026,
        "flow_mse": 1.1201456785202026,
        "t_mean": 0.5838073492050171,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96875,
        "ot_gate_min": 0.9296875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 43,
        "loss": 1.039523959159851,
        "flow_mse": 1.039523959159851,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97216796875,
        "ot_gate_min": 0.94091796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 44,
        "loss": 1.0210646390914917,
        "flow_mse": 1.0210646390914917,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9677734375,
        "ot_gate_min": 0.93798828125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 45,
        "loss": 1.0069496631622314,
        "flow_mse": 1.0069496631622314,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.974609375,
        "ot_gate_min": 0.9482421875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 46,
        "loss": 1.0390965938568115,
        "flow_mse": 1.0390965938568115,
        "t_mean": 0.3058706820011139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9697265625,
        "ot_gate_min": 0.9296875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 47,
        "loss": 1.0197845697402954,
        "flow_mse": 1.0197845697402954,
        "t_mean": 0.31310853362083435,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97021484375,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 48,
        "loss": 1.0371644496917725,
        "flow_mse": 1.0371644496917725,
        "t_mean": 0.5626304149627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97412109375,
        "ot_gate_min": 0.9423828125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 49,
        "loss": 1.036482810974121,
        "flow_mse": 1.036482810974121,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9736328125,
        "ot_gate_min": 0.94091796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 50,
        "loss": 1.0249230861663818,
        "flow_mse": 1.0249230861663818,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9716796875,
        "ot_gate_min": 0.94091796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 51,
        "loss": 1.0295422077178955,
        "flow_mse": 1.0295422077178955,
        "t_mean": 0.6815006732940674,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97021484375,
        "ot_gate_min": 0.93701171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 52,
        "loss": 1.0306437015533447,
        "flow_mse": 1.0306437015533447,
        "t_mean": 0.509903609752655,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97216796875,
        "ot_gate_min": 0.93701171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 53,
        "loss": 1.016850233078003,
        "flow_mse": 1.016850233078003,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97021484375,
        "ot_gate_min": 0.93212890625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 54,
        "loss": 1.0910499095916748,
        "flow_mse": 1.0910499095916748,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96826171875,
        "ot_gate_min": 0.9296875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 55,
        "loss": 1.052876353263855,
        "flow_mse": 1.052876353263855,
        "t_mean": 0.7171943187713623,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9736328125,
        "ot_gate_min": 0.9423828125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 56,
        "loss": 1.0530800819396973,
        "flow_mse": 1.0530800819396973,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97216796875,
        "ot_gate_min": 0.93896484375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 57,
        "loss": 1.0307668447494507,
        "flow_mse": 1.0307668447494507,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97509765625,
        "ot_gate_min": 0.94921875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 58,
        "loss": 1.0234510898590088,
        "flow_mse": 1.0234510898590088,
        "t_mean": 0.5147815942764282,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9765625,
        "ot_gate_min": 0.95166015625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 59,
        "loss": 1.0753936767578125,
        "flow_mse": 1.0753936767578125,
        "t_mean": 0.49480095505714417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97509765625,
        "ot_gate_min": 0.94921875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 60,
        "loss": 1.0311006307601929,
        "flow_mse": 1.0311006307601929,
        "t_mean": 0.4051949977874756,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.970703125,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Full Stage1 mask prior",
    "group": "stage1",
    "phase": "full",
    "note": "Full matrix with mask labels as a soft OT prior.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage1_mask_prior",
    "status": "ok",
    "steps": 60,
    "loss": 1.0311006307601929,
    "flow_mse": 1.0311006307601929,
    "art_l2": null,
    "ot_gate_mean": 0.27099609375,
    "ot_edge_delta": 0.0,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage1_mask_prior",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2919921875,
        "ot_gate_min": 0.0208587646484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.19677734375,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.36572265625,
        "ot_gate_min": 0.0225067138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.267822265625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.230224609375,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2335205078125,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.336181640625,
        "ot_gate_min": 0.0218505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32373046875,
        "ot_gate_min": 0.02618408203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 1.0248684883117676,
        "flow_mse": 1.0248684883117676,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2344970703125,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 1.0194365978240967,
        "flow_mse": 1.0194365978240967,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26171875,
        "ot_gate_min": 0.0212554931640625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 1.059285044670105,
        "flow_mse": 1.059285044670105,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.361328125,
        "ot_gate_min": 0.0225067138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 1.030677318572998,
        "flow_mse": 1.030677318572998,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3896484375,
        "ot_gate_min": 0.02264404296875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 1.0135877132415771,
        "flow_mse": 1.0135877132415771,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.37841796875,
        "ot_gate_min": 0.02215576171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 1.0318082571029663,
        "flow_mse": 1.0318082571029663,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2440185546875,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 1.027304768562317,
        "flow_mse": 1.027304768562317,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.15478515625,
        "ot_gate_min": 0.02166748046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 1.0195131301879883,
        "flow_mse": 1.0195131301879883,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1397705078125,
        "ot_gate_min": 0.02191162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 1.0307263135910034,
        "flow_mse": 1.0307263135910034,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.257080078125,
        "ot_gate_min": 0.0212249755859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 1.0178287029266357,
        "flow_mse": 1.0178287029266357,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.141357421875,
        "ot_gate_min": 0.0218963623046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 1.0279525518417358,
        "flow_mse": 1.0279525518417358,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1573486328125,
        "ot_gate_min": 0.0225830078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 1.03252112865448,
        "flow_mse": 1.03252112865448,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1885986328125,
        "ot_gate_min": 0.021575927734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 21,
        "loss": 1.0540964603424072,
        "flow_mse": 1.0540964603424072,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2529296875,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 22,
        "loss": 1.0650438070297241,
        "flow_mse": 1.0650438070297241,
        "t_mean": 0.5307409167289734,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3486328125,
        "ot_gate_min": 0.0259552001953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 23,
        "loss": 1.0473723411560059,
        "flow_mse": 1.0473723411560059,
        "t_mean": 0.511449933052063,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2861328125,
        "ot_gate_min": 0.025726318359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 24,
        "loss": 1.0280429124832153,
        "flow_mse": 1.0280429124832153,
        "t_mean": 0.5524937510490417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.266845703125,
        "ot_gate_min": 0.0217132568359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 25,
        "loss": 1.0299099683761597,
        "flow_mse": 1.0299099683761597,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2445068359375,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 26,
        "loss": 1.0370187759399414,
        "flow_mse": 1.0370187759399414,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.397216796875,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 27,
        "loss": 1.0062942504882812,
        "flow_mse": 1.0062942504882812,
        "t_mean": 0.7675442695617676,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32421875,
        "ot_gate_min": 0.0239105224609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 28,
        "loss": 1.099644660949707,
        "flow_mse": 1.099644660949707,
        "t_mean": 0.6012675166130066,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.22509765625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 29,
        "loss": 1.0200495719909668,
        "flow_mse": 1.0200495719909668,
        "t_mean": 0.6016367673873901,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.263427734375,
        "ot_gate_min": 0.021728515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 30,
        "loss": 1.0361602306365967,
        "flow_mse": 1.0361602306365967,
        "t_mean": 0.5350275039672852,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1766357421875,
        "ot_gate_min": 0.0222930908203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 31,
        "loss": 1.023235559463501,
        "flow_mse": 1.023235559463501,
        "t_mean": 0.3996628522872925,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283203125,
        "ot_gate_min": 0.021087646484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 32,
        "loss": 1.0194340944290161,
        "flow_mse": 1.0194340944290161,
        "t_mean": 0.4242017865180969,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.271484375,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 33,
        "loss": 1.0286738872528076,
        "flow_mse": 1.0286738872528076,
        "t_mean": 0.6198998689651489,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3916015625,
        "ot_gate_min": 0.0218048095703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 34,
        "loss": 1.0266056060791016,
        "flow_mse": 1.0266056060791016,
        "t_mean": 0.4256276786327362,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.28369140625,
        "ot_gate_min": 0.0211029052734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 35,
        "loss": 1.021877646446228,
        "flow_mse": 1.021877646446228,
        "t_mean": 0.5468468070030212,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.179931640625,
        "ot_gate_min": 0.021240234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 36,
        "loss": 1.035739779472351,
        "flow_mse": 1.035739779472351,
        "t_mean": 0.4543689489364624,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1873779296875,
        "ot_gate_min": 0.0213775634765625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 37,
        "loss": 1.0911012887954712,
        "flow_mse": 1.0911012887954712,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.224853515625,
        "ot_gate_min": 0.0219573974609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 38,
        "loss": 1.0154587030410767,
        "flow_mse": 1.0154587030410767,
        "t_mean": 0.4061962068080902,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.275390625,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 39,
        "loss": 1.030579924583435,
        "flow_mse": 1.030579924583435,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25,
        "ot_gate_min": 0.02191162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 40,
        "loss": 1.041369080543518,
        "flow_mse": 1.041369080543518,
        "t_mean": 0.4235572814941406,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.400634765625,
        "ot_gate_min": 0.021728515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 41,
        "loss": 1.0356323719024658,
        "flow_mse": 1.0356323719024658,
        "t_mean": 0.3948693871498108,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.35986328125,
        "ot_gate_min": 0.0227813720703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 42,
        "loss": 1.1201456785202026,
        "flow_mse": 1.1201456785202026,
        "t_mean": 0.5838073492050171,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.269287109375,
        "ot_gate_min": 0.021514892578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 43,
        "loss": 1.039523959159851,
        "flow_mse": 1.039523959159851,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.271484375,
        "ot_gate_min": 0.021026611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 44,
        "loss": 1.0210646390914917,
        "flow_mse": 1.0210646390914917,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26513671875,
        "ot_gate_min": 0.02117919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 45,
        "loss": 1.0069496631622314,
        "flow_mse": 1.0069496631622314,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.37353515625,
        "ot_gate_min": 0.0218048095703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 46,
        "loss": 1.0390965938568115,
        "flow_mse": 1.0390965938568115,
        "t_mean": 0.3058706820011139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.19189453125,
        "ot_gate_min": 0.0215301513671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 47,
        "loss": 1.0197845697402954,
        "flow_mse": 1.0197845697402954,
        "t_mean": 0.31310853362083435,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.15576171875,
        "ot_gate_min": 0.0217132568359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 48,
        "loss": 1.0371644496917725,
        "flow_mse": 1.0371644496917725,
        "t_mean": 0.5626304149627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.392578125,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 49,
        "loss": 1.036482810974121,
        "flow_mse": 1.036482810974121,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.28173828125,
        "ot_gate_min": 0.02099609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 50,
        "loss": 1.0249230861663818,
        "flow_mse": 1.0249230861663818,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.275634765625,
        "ot_gate_min": 0.021026611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 51,
        "loss": 1.0295422077178955,
        "flow_mse": 1.0295422077178955,
        "t_mean": 0.6815006732940674,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.388916015625,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 52,
        "loss": 1.0306437015533447,
        "flow_mse": 1.0306437015533447,
        "t_mean": 0.509903609752655,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39111328125,
        "ot_gate_min": 0.0228424072265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 53,
        "loss": 1.016850233078003,
        "flow_mse": 1.016850233078003,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39404296875,
        "ot_gate_min": 0.0220184326171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 54,
        "loss": 1.0910499095916748,
        "flow_mse": 1.0910499095916748,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.22509765625,
        "ot_gate_min": 0.0219268798828125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 55,
        "loss": 1.052876353263855,
        "flow_mse": 1.052876353263855,
        "t_mean": 0.7171943187713623,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25244140625,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 56,
        "loss": 1.0530800819396973,
        "flow_mse": 1.0530800819396973,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.256591796875,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 57,
        "loss": 1.0307668447494507,
        "flow_mse": 1.0307668447494507,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39794921875,
        "ot_gate_min": 0.0235748291015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 58,
        "loss": 1.0234510898590088,
        "flow_mse": 1.0234510898590088,
        "t_mean": 0.5147815942764282,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29052734375,
        "ot_gate_min": 0.0209503173828125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 59,
        "loss": 1.075393557548523,
        "flow_mse": 1.075393557548523,
        "t_mean": 0.49480095505714417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.292724609375,
        "ot_gate_min": 0.025238037109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 60,
        "loss": 1.0311006307601929,
        "flow_mse": 1.0311006307601929,
        "t_mean": 0.4051949977874756,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.27099609375,
        "ot_gate_min": 0.021575927734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Full Stage1 RGB edge",
    "group": "stage1",
    "phase": "full",
    "note": "Full matrix with RGB Sobel edge-regularized OT cost.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage1_rgb_edge",
    "status": "ok",
    "steps": 60,
    "loss": 1.0311006307601929,
    "flow_mse": 1.0311006307601929,
    "art_l2": null,
    "ot_gate_mean": 0.271240234375,
    "ot_edge_delta": 0.001356494496576488,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage1_rgb_edge",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012463603634387255,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2919921875,
        "ot_gate_min": 0.0208587646484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013847764348611236,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1971435546875,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001398036372847855,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3662109375,
        "ot_gate_min": 0.02252197265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001753590302541852,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.267822265625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013486214447766542,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.23046875,
        "ot_gate_min": 0.02178955078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001678119762800634,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2340087890625,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014689782401546836,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.336669921875,
        "ot_gate_min": 0.0218505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001582509372383356,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32373046875,
        "ot_gate_min": 0.0261993408203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 1.0248684883117676,
        "flow_mse": 1.0248684883117676,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0018019641283899546,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2349853515625,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 1.0194365978240967,
        "flow_mse": 1.0194365978240967,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015105771599337459,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26171875,
        "ot_gate_min": 0.0212554931640625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 1.059285044670105,
        "flow_mse": 1.059285044670105,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015028680209070444,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.36181640625,
        "ot_gate_min": 0.0225067138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 1.030677318572998,
        "flow_mse": 1.030677318572998,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013878400204703212,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.389892578125,
        "ot_gate_min": 0.0226593017578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 1.0135877132415771,
        "flow_mse": 1.0135877132415771,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014244270278140903,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.37841796875,
        "ot_gate_min": 0.02215576171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 1.0318082571029663,
        "flow_mse": 1.0318082571029663,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014156353427097201,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2445068359375,
        "ot_gate_min": 0.0214385986328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 1.027304768562317,
        "flow_mse": 1.027304768562317,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016134991310536861,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1549072265625,
        "ot_gate_min": 0.0216827392578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 1.0195131301879883,
        "flow_mse": 1.0195131301879883,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014201640151441097,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.139892578125,
        "ot_gate_min": 0.02191162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 1.0307263135910034,
        "flow_mse": 1.0307263135910034,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014097682433202863,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.257080078125,
        "ot_gate_min": 0.0212249755859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 1.0178287029266357,
        "flow_mse": 1.0178287029266357,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015146248042583466,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.141357421875,
        "ot_gate_min": 0.0218963623046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 1.0279525518417358,
        "flow_mse": 1.0279525518417358,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012146909721195698,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1573486328125,
        "ot_gate_min": 0.0225830078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 1.03252112865448,
        "flow_mse": 1.03252112865448,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016136284684762359,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1885986328125,
        "ot_gate_min": 0.021575927734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 21,
        "loss": 1.0540964603424072,
        "flow_mse": 1.0540964603424072,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015243198722600937,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2529296875,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 22,
        "loss": 1.0650438070297241,
        "flow_mse": 1.0650438070297241,
        "t_mean": 0.5307409167289734,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001435597543604672,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3486328125,
        "ot_gate_min": 0.0259552001953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 23,
        "loss": 1.0473723411560059,
        "flow_mse": 1.0473723411560059,
        "t_mean": 0.511449933052063,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011276646982878447,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2861328125,
        "ot_gate_min": 0.0257415771484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 24,
        "loss": 1.0280429124832153,
        "flow_mse": 1.0280429124832153,
        "t_mean": 0.5524937510490417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013031219132244587,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26708984375,
        "ot_gate_min": 0.0217132568359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 25,
        "loss": 1.0299099683761597,
        "flow_mse": 1.0299099683761597,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013676118105649948,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2445068359375,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 26,
        "loss": 1.0370187759399414,
        "flow_mse": 1.0370187759399414,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016084759263321757,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39794921875,
        "ot_gate_min": 0.0217132568359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 27,
        "loss": 1.0062942504882812,
        "flow_mse": 1.0062942504882812,
        "t_mean": 0.7675442695617676,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011069304309785366,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32421875,
        "ot_gate_min": 0.0239105224609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 28,
        "loss": 1.099644660949707,
        "flow_mse": 1.099644660949707,
        "t_mean": 0.6012675166130066,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012524642515927553,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.22509765625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 29,
        "loss": 1.0200495719909668,
        "flow_mse": 1.0200495719909668,
        "t_mean": 0.6016367673873901,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001515839365310967,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.263916015625,
        "ot_gate_min": 0.021728515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 30,
        "loss": 1.0361602306365967,
        "flow_mse": 1.0361602306365967,
        "t_mean": 0.5350275039672852,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001293253735639155,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1768798828125,
        "ot_gate_min": 0.0222930908203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 31,
        "loss": 1.023235559463501,
        "flow_mse": 1.023235559463501,
        "t_mean": 0.3996628522872925,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012450056383386254,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283203125,
        "ot_gate_min": 0.021087646484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 32,
        "loss": 1.0194340944290161,
        "flow_mse": 1.0194340944290161,
        "t_mean": 0.4242017865180969,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0018455247627571225,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.271728515625,
        "ot_gate_min": 0.0214385986328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 33,
        "loss": 1.0286738872528076,
        "flow_mse": 1.0286738872528076,
        "t_mean": 0.6198998689651489,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001384816481731832,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.392333984375,
        "ot_gate_min": 0.0218048095703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 34,
        "loss": 1.0266056060791016,
        "flow_mse": 1.0266056060791016,
        "t_mean": 0.4256276786327362,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001197467907331884,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.28369140625,
        "ot_gate_min": 0.0211029052734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 35,
        "loss": 1.021877646446228,
        "flow_mse": 1.021877646446228,
        "t_mean": 0.5468468070030212,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015151776606217027,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.180419921875,
        "ot_gate_min": 0.021240234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 36,
        "loss": 1.035739779472351,
        "flow_mse": 1.035739779472351,
        "t_mean": 0.4543689489364624,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001779106562025845,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1878662109375,
        "ot_gate_min": 0.0213775634765625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 37,
        "loss": 1.0911012887954712,
        "flow_mse": 1.0911012887954712,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012280209921300411,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2249755859375,
        "ot_gate_min": 0.0219573974609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 38,
        "loss": 1.0154587030410767,
        "flow_mse": 1.0154587030410767,
        "t_mean": 0.4061962068080902,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0020692418329417706,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.27587890625,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 39,
        "loss": 1.030579924583435,
        "flow_mse": 1.030579924583435,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014930220786482096,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25,
        "ot_gate_min": 0.02191162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 40,
        "loss": 1.041369080543518,
        "flow_mse": 1.041369080543518,
        "t_mean": 0.4235572814941406,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016851616092026234,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.401123046875,
        "ot_gate_min": 0.021728515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 41,
        "loss": 1.0356323719024658,
        "flow_mse": 1.0356323719024658,
        "t_mean": 0.3948693871498108,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010541785741224885,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.35986328125,
        "ot_gate_min": 0.0227813720703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 42,
        "loss": 1.1201456785202026,
        "flow_mse": 1.1201456785202026,
        "t_mean": 0.5838073492050171,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0019060715567320585,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.269775390625,
        "ot_gate_min": 0.021514892578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 43,
        "loss": 1.039523959159851,
        "flow_mse": 1.039523959159851,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001424789079464972,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.27197265625,
        "ot_gate_min": 0.021026611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 44,
        "loss": 1.0210646390914917,
        "flow_mse": 1.0210646390914917,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014364580856636167,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26513671875,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 45,
        "loss": 1.0069496631622314,
        "flow_mse": 1.0069496631622314,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014721160987392068,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.37353515625,
        "ot_gate_min": 0.0218048095703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 46,
        "loss": 1.0390965938568115,
        "flow_mse": 1.0390965938568115,
        "t_mean": 0.3058706820011139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016219338867813349,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.192138671875,
        "ot_gate_min": 0.0215301513671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 47,
        "loss": 1.0197845697402954,
        "flow_mse": 1.0197845697402954,
        "t_mean": 0.31310853362083435,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001696025487035513,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.15576171875,
        "ot_gate_min": 0.0217132568359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 48,
        "loss": 1.0371644496917725,
        "flow_mse": 1.0371644496917725,
        "t_mean": 0.5626304149627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009278897778131068,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.392578125,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 49,
        "loss": 1.036482810974121,
        "flow_mse": 1.036482810974121,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015015194658190012,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.28173828125,
        "ot_gate_min": 0.02099609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 50,
        "loss": 1.0249230861663818,
        "flow_mse": 1.0249230861663818,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015260844957083464,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.275634765625,
        "ot_gate_min": 0.021026611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 51,
        "loss": 1.0295422077178955,
        "flow_mse": 1.0295422077178955,
        "t_mean": 0.6815006732940674,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016038741450756788,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.388916015625,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 52,
        "loss": 1.0306437015533447,
        "flow_mse": 1.0306437015533447,
        "t_mean": 0.509903609752655,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001483110012486577,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.391357421875,
        "ot_gate_min": 0.022857666015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 53,
        "loss": 1.016850233078003,
        "flow_mse": 1.016850233078003,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016984629910439253,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.394287109375,
        "ot_gate_min": 0.02203369140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 54,
        "loss": 1.0910499095916748,
        "flow_mse": 1.0910499095916748,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013595152413472533,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.22509765625,
        "ot_gate_min": 0.021942138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 55,
        "loss": 1.052876353263855,
        "flow_mse": 1.052876353263855,
        "t_mean": 0.7171943187713623,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001471476862207055,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.252685546875,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 56,
        "loss": 1.0530800819396973,
        "flow_mse": 1.0530800819396973,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015233580488711596,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2568359375,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 57,
        "loss": 1.0307668447494507,
        "flow_mse": 1.0307668447494507,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014293794520199299,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39794921875,
        "ot_gate_min": 0.0235748291015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 58,
        "loss": 1.0234510898590088,
        "flow_mse": 1.0234510898590088,
        "t_mean": 0.5147815942764282,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015365192666649818,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.290771484375,
        "ot_gate_min": 0.0209503173828125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 59,
        "loss": 1.075393557548523,
        "flow_mse": 1.075393557548523,
        "t_mean": 0.49480095505714417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011814767494797707,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29296875,
        "ot_gate_min": 0.0252532958984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 60,
        "loss": 1.0311006307601929,
        "flow_mse": 1.0311006307601929,
        "t_mean": 0.4051949977874756,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001356494496576488,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.271240234375,
        "ot_gate_min": 0.0215911865234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Full Stage1 virtual patch",
    "group": "stage1",
    "phase": "full",
    "note": "Full matrix with a virtual patch channel.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage1_virtual_patch",
    "status": "ok",
    "steps": 60,
    "loss": 1.0239849090576172,
    "flow_mse": 1.0239849090576172,
    "art_l2": null,
    "ot_gate_mean": 0.2900390625,
    "ot_edge_delta": 0.001642722636461258,
    "ot_virtual_mass": 0.04998779296875,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "virtual_tokens": 1,
      "virtual_mass": 0.05,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage1_virtual_patch",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_virtual_tokens": 1,
      "ot_virtual_mass": 0.05,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0175116062164307,
        "flow_mse": 1.0175116062164307,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013981556985527277,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.400634765625,
        "ot_gate_min": 0.0216217041015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 2,
        "loss": 1.0188701152801514,
        "flow_mse": 1.0188701152801514,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013099866919219494,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2154541015625,
        "ot_gate_min": 0.022247314453125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 3,
        "loss": 1.0372271537780762,
        "flow_mse": 1.0372271537780762,
        "t_mean": 0.6866899728775024,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013684009900316596,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.15869140625,
        "ot_gate_min": 0.0219573974609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 4,
        "loss": 1.0259208679199219,
        "flow_mse": 1.0259208679199219,
        "t_mean": 0.5447205901145935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001627056859433651,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.158203125,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 5,
        "loss": 1.0232212543487549,
        "flow_mse": 1.0232212543487549,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015082642203196883,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.260986328125,
        "ot_gate_min": 0.0213623046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 6,
        "loss": 1.0559325218200684,
        "flow_mse": 1.0559325218200684,
        "t_mean": 0.681830883026123,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012014677049592137,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29248046875,
        "ot_gate_min": 0.0252532958984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 7,
        "loss": 1.0603113174438477,
        "flow_mse": 1.0603113174438477,
        "t_mean": 0.7412480115890503,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014335500309243798,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.349365234375,
        "ot_gate_min": 0.0259857177734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 8,
        "loss": 1.033375859260559,
        "flow_mse": 1.033375859260559,
        "t_mean": 0.39051905274391174,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014481099788099527,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.146240234375,
        "ot_gate_min": 0.021942138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 9,
        "loss": 1.0245479345321655,
        "flow_mse": 1.0245479345321655,
        "t_mean": 0.5102756023406982,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.002222604351118207,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.167724609375,
        "ot_gate_min": 0.0216217041015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 10,
        "loss": 1.1110031604766846,
        "flow_mse": 1.1110031604766846,
        "t_mean": 0.5520592927932739,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0019456963054835796,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.266845703125,
        "ot_gate_min": 0.021728515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 11,
        "loss": 1.0233906507492065,
        "flow_mse": 1.0233906507492065,
        "t_mean": 0.593870222568512,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.00211955513805151,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.27392578125,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 12,
        "loss": 1.0197861194610596,
        "flow_mse": 1.0197861194610596,
        "t_mean": 0.42200398445129395,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013815348502248526,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.37744140625,
        "ot_gate_min": 0.0223846435546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 13,
        "loss": 1.0211420059204102,
        "flow_mse": 1.0211420059204102,
        "t_mean": 0.39839279651641846,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001588745741173625,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.264404296875,
        "ot_gate_min": 0.0214996337890625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 14,
        "loss": 1.0618864297866821,
        "flow_mse": 1.0618864297866821,
        "t_mean": 0.581339418888092,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001437863102182746,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.381103515625,
        "ot_gate_min": 0.024444580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 15,
        "loss": 1.0441529750823975,
        "flow_mse": 1.0441529750823975,
        "t_mean": 0.3876829147338867,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016030267579481006,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.392578125,
        "ot_gate_min": 0.0226287841796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 16,
        "loss": 1.0604774951934814,
        "flow_mse": 1.0604774951934814,
        "t_mean": 0.5303117036819458,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015838773688301444,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.324462890625,
        "ot_gate_min": 0.026214599609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 17,
        "loss": 1.0307263135910034,
        "flow_mse": 1.0307263135910034,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015775307547301054,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26513671875,
        "ot_gate_min": 0.021209716796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 18,
        "loss": 1.0296268463134766,
        "flow_mse": 1.0296268463134766,
        "t_mean": 0.4426919221878052,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015036083059385419,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.172119140625,
        "ot_gate_min": 0.022064208984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 19,
        "loss": 1.0364758968353271,
        "flow_mse": 1.0364758968353271,
        "t_mean": 0.4035862386226654,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009270749869756401,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.393310546875,
        "ot_gate_min": 0.023040771484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 20,
        "loss": 1.0321329832077026,
        "flow_mse": 1.0321329832077026,
        "t_mean": 0.52027428150177,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016400060849264264,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26123046875,
        "ot_gate_min": 0.021270751953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 21,
        "loss": 1.0356372594833374,
        "flow_mse": 1.0356372594833374,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014349027769640088,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2646484375,
        "ot_gate_min": 0.021270751953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 22,
        "loss": 1.0323278903961182,
        "flow_mse": 1.0323278903961182,
        "t_mean": 0.5029113292694092,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014791389694437385,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.161865234375,
        "ot_gate_min": 0.0215911865234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 23,
        "loss": 1.0173629522323608,
        "flow_mse": 1.0173629522323608,
        "t_mean": 0.45976296067237854,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001550937071442604,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.262451171875,
        "ot_gate_min": 0.0218353271484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 24,
        "loss": 1.0596609115600586,
        "flow_mse": 1.0596609115600586,
        "t_mean": 0.7851753234863281,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013905689120292664,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.323974609375,
        "ot_gate_min": 0.02606201171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 25,
        "loss": 1.029831886291504,
        "flow_mse": 1.029831886291504,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001520897843874991,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26611328125,
        "ot_gate_min": 0.021484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 26,
        "loss": 1.0282132625579834,
        "flow_mse": 1.0282132625579834,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.00135008932556957,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2313232421875,
        "ot_gate_min": 0.021881103515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 27,
        "loss": 1.0246374607086182,
        "flow_mse": 1.0246374607086182,
        "t_mean": 0.5499228239059448,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013658726820722222,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.244384765625,
        "ot_gate_min": 0.021514892578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 28,
        "loss": 1.053013801574707,
        "flow_mse": 1.053013801574707,
        "t_mean": 0.5113206505775452,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015291013987734914,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.33740234375,
        "ot_gate_min": 0.0239410400390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 29,
        "loss": 1.031076431274414,
        "flow_mse": 1.031076431274414,
        "t_mean": 0.5222197771072388,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016025243094190955,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.389404296875,
        "ot_gate_min": 0.0230255126953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 30,
        "loss": 1.0421171188354492,
        "flow_mse": 1.0421171188354492,
        "t_mean": 0.5769186019897461,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014842785894870758,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2587890625,
        "ot_gate_min": 0.02130126953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 31,
        "loss": 1.0545744895935059,
        "flow_mse": 1.0545744895935059,
        "t_mean": 0.314164400100708,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001457578269764781,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.388427734375,
        "ot_gate_min": 0.0230255126953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 32,
        "loss": 1.0230286121368408,
        "flow_mse": 1.0230286121368408,
        "t_mean": 0.5159284472465515,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016797517891973257,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.272705078125,
        "ot_gate_min": 0.021270751953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 33,
        "loss": 1.0242466926574707,
        "flow_mse": 1.0242466926574707,
        "t_mean": 0.5761235952377319,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015731928870081902,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26171875,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 34,
        "loss": 1.0617096424102783,
        "flow_mse": 1.0617096424102783,
        "t_mean": 0.3432917296886444,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015714274486526847,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3974609375,
        "ot_gate_min": 0.0254974365234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 35,
        "loss": 1.0548802614212036,
        "flow_mse": 1.0548802614212036,
        "t_mean": 0.5136957168579102,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015504645416513085,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.384765625,
        "ot_gate_min": 0.024169921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 36,
        "loss": 1.0237579345703125,
        "flow_mse": 1.0237579345703125,
        "t_mean": 0.449867308139801,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015528254443779588,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.269287109375,
        "ot_gate_min": 0.02166748046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 37,
        "loss": 1.0211809873580933,
        "flow_mse": 1.0211809873580933,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015051080845296383,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.260498046875,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 38,
        "loss": 1.0251076221466064,
        "flow_mse": 1.0251076221466064,
        "t_mean": 0.49369919300079346,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0018110070377588272,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.18994140625,
        "ot_gate_min": 0.0215301513671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 39,
        "loss": 1.0227961540222168,
        "flow_mse": 1.0227961540222168,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012435183161869645,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.282470703125,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 40,
        "loss": 1.0196993350982666,
        "flow_mse": 1.0196993350982666,
        "t_mean": 0.5427849888801575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016729397466406226,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.37060546875,
        "ot_gate_min": 0.0220794677734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 41,
        "loss": 1.02742600440979,
        "flow_mse": 1.02742600440979,
        "t_mean": 0.6182063817977905,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015907728811725974,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2335205078125,
        "ot_gate_min": 0.0218353271484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 42,
        "loss": 1.0230839252471924,
        "flow_mse": 1.0230839252471924,
        "t_mean": 0.5307261943817139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0017388258129358292,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1339111328125,
        "ot_gate_min": 0.02191162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 43,
        "loss": 1.0291386842727661,
        "flow_mse": 1.0291386842727661,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013169687008485198,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.27490234375,
        "ot_gate_min": 0.0210418701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 44,
        "loss": 1.0190776586532593,
        "flow_mse": 1.0190776586532593,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001553501351736486,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25244140625,
        "ot_gate_min": 0.0214385986328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 45,
        "loss": 1.0321238040924072,
        "flow_mse": 1.0321238040924072,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013981072697788477,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.365234375,
        "ot_gate_min": 0.0226593017578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 46,
        "loss": 1.0276533365249634,
        "flow_mse": 1.0276533365249634,
        "t_mean": 0.3461093604564667,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0017451198073104024,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.158447265625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 47,
        "loss": 1.0564388036727905,
        "flow_mse": 1.0564388036727905,
        "t_mean": 0.370074987411499,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014228996587917209,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.255859375,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 48,
        "loss": 1.041345477104187,
        "flow_mse": 1.041345477104187,
        "t_mean": 0.6085288524627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001981860725209117,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.164794921875,
        "ot_gate_min": 0.0218353271484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 49,
        "loss": 1.1018879413604736,
        "flow_mse": 1.1018879413604736,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001291007036343217,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2210693359375,
        "ot_gate_min": 0.0224456787109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 50,
        "loss": 1.0263246297836304,
        "flow_mse": 1.0263246297836304,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014104365836828947,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.20458984375,
        "ot_gate_min": 0.022735595703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 51,
        "loss": 1.0363283157348633,
        "flow_mse": 1.0363283157348633,
        "t_mean": 0.5388283729553223,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0017775115557014942,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.18798828125,
        "ot_gate_min": 0.021453857421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 52,
        "loss": 1.061694622039795,
        "flow_mse": 1.061694622039795,
        "t_mean": 0.5041043758392334,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001250310568138957,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29345703125,
        "ot_gate_min": 0.025299072265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 53,
        "loss": 1.0269901752471924,
        "flow_mse": 1.0269901752471924,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009288513683713973,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.20166015625,
        "ot_gate_min": 0.0228118896484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 54,
        "loss": 1.027586817741394,
        "flow_mse": 1.027586817741394,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0015298461075872183,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.264892578125,
        "ot_gate_min": 0.02130126953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 55,
        "loss": 1.0603880882263184,
        "flow_mse": 1.0603880882263184,
        "t_mean": 0.6222312450408936,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014486979926005006,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.35107421875,
        "ot_gate_min": 0.022735595703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 56,
        "loss": 1.038100004196167,
        "flow_mse": 1.038100004196167,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001455579069443047,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.4072265625,
        "ot_gate_min": 0.0213165283203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 57,
        "loss": 1.0684614181518555,
        "flow_mse": 1.0684614181518555,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0018026758916676044,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.38671875,
        "ot_gate_min": 0.0243988037109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 58,
        "loss": 1.0254374742507935,
        "flow_mse": 1.0254374742507935,
        "t_mean": 0.4430321455001831,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014936915831640363,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2330322265625,
        "ot_gate_min": 0.021881103515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 59,
        "loss": 1.0593796968460083,
        "flow_mse": 1.0593796968460083,
        "t_mean": 0.45887431502342224,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014710568357259035,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25244140625,
        "ot_gate_min": 0.0215911865234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 60,
        "loss": 1.0239849090576172,
        "flow_mse": 1.0239849090576172,
        "t_mean": 0.46173495054244995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001642722636461258,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2900390625,
        "ot_gate_min": 0.021026611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      }
    ]
  },
  {
    "label": "Full Stage1 first-third edge",
    "group": "stage1",
    "phase": "full",
    "note": "Full matrix with OT injected into the first third of Stage1 blocks.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage1_firstthird_edge",
    "status": "ok",
    "steps": 20,
    "loss": 1.0325210094451904,
    "flow_mse": 1.0325210094451904,
    "art_l2": null,
    "ot_gate_mean": 0.1809234619140625,
    "ot_edge_delta": 0.0020620503782993183,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 8.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first-third",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage1_firstthird_edge",
      "finetune_ckpt": null,
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 20,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 1.0163291692733765,
        "flow_mse": 1.0163291692733765,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001615208646398969,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.280731201171875,
        "ot_gate_min": 0.020847320556640625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 1.0261752605438232,
        "flow_mse": 1.0261752605438232,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001815385592635721,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.195343017578125,
        "ot_gate_min": 0.022920608520507812,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 1.0302751064300537,
        "flow_mse": 1.0302751064300537,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0018975692219100893,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.36029052734375,
        "ot_gate_min": 0.022266387939453125,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 1.0295436382293701,
        "flow_mse": 1.0295436382293701,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.002060391299892217,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.258026123046875,
        "ot_gate_min": 0.021093368530273438,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 1.0270346403121948,
        "flow_mse": 1.0270346403121948,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0017479378002462909,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.222015380859375,
        "ot_gate_min": 0.021474838256835938,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 1.0205926895141602,
        "flow_mse": 1.0205926895141602,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001995620463276282,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.20025634765625,
        "ot_gate_min": 0.02146148681640625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 1.0244700908660889,
        "flow_mse": 1.0244700908660889,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0020024599216412753,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.291351318359375,
        "ot_gate_min": 0.021503448486328125,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 1.0630731582641602,
        "flow_mse": 1.0630731582641602,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0017050473397830501,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.320526123046875,
        "ot_gate_min": 0.02542877197265625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 1.0248684883117676,
        "flow_mse": 1.0248684883117676,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.00215379917062819,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.201812744140625,
        "ot_gate_min": 0.021381378173828125,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 1.0194365978240967,
        "flow_mse": 1.0194365978240967,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0017591614450793713,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25140380859375,
        "ot_gate_min": 0.021226882934570312,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 1.059285044670105,
        "flow_mse": 1.059285044670105,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0018290871666977182,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.356201171875,
        "ot_gate_min": 0.022100448608398438,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 1.0306774377822876,
        "flow_mse": 1.0306774377822876,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0016916544700507075,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.36602783203125,
        "ot_gate_min": 0.022281646728515625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 1.0135877132415771,
        "flow_mse": 1.0135877132415771,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001653539904509671,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.37005615234375,
        "ot_gate_min": 0.0218658447265625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 1.0318082571029663,
        "flow_mse": 1.0318082571029663,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0017324336804449558,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.236480712890625,
        "ot_gate_min": 0.021389007568359375,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 1.027304768562317,
        "flow_mse": 1.027304768562317,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0019666101288748905,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1427459716796875,
        "ot_gate_min": 0.021448135375976562,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 1.0195131301879883,
        "flow_mse": 1.0195131301879883,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0018350208993069828,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1317901611328125,
        "ot_gate_min": 0.021762847900390625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 1.0307263135910034,
        "flow_mse": 1.0307263135910034,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0018481845472706482,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2457427978515625,
        "ot_gate_min": 0.020999908447265625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 1.0178287029266357,
        "flow_mse": 1.0178287029266357,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0019221086840843782,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1320343017578125,
        "ot_gate_min": 0.0214996337890625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 1.0279525518417358,
        "flow_mse": 1.0279525518417358,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0016005436627892777,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1532135009765625,
        "ot_gate_min": 0.022182464599609375,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 1.0325210094451904,
        "flow_mse": 1.0325210094451904,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0020620503782993183,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1809234619140625,
        "ot_gate_min": 0.021190643310546875,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Full Stage2 spatial diag",
    "group": "stage2",
    "phase": "full",
    "note": "Full matrix Stage2 spatial diagnostics.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage2_spatial_diag",
    "status": "ok",
    "steps": 40,
    "loss": 1.5618746280670166,
    "flow_mse": 1.4998279809951782,
    "art_l2": 0.06204666197299957,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 10.0,
    "stage2_mask_empty_parts": 3.0,
    "stage2_mask_centroid_spread": 0.12488780915737152,
    "stage2_mask_spatial_var_mean": 0.019491467624902725,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage2_spatial_diag",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 40,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 2.38563871383667,
        "flow_mse": 2.275019884109497,
        "art_l2": 0.11061874032020569,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 5.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.097297303378582,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1345607340335846,
        "stage2_mask_spatial_var_mean": 0.03559853881597519
      },
      {
        "step": 2,
        "loss": 1.0193780660629272,
        "flow_mse": 0.934458315372467,
        "art_l2": 0.08491974323987961,
        "t_mean": 0.3323965072631836,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.07617656141519547,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11268322169780731,
        "stage2_mask_spatial_var_mean": 0.024656636640429497
      },
      {
        "step": 3,
        "loss": 4.098177433013916,
        "flow_mse": 4.016501426696777,
        "art_l2": 0.08167603611946106,
        "t_mean": 0.8121371269226074,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.1260043829679489,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13602715730667114,
        "stage2_mask_spatial_var_mean": 0.055630188435316086
      },
      {
        "step": 4,
        "loss": 2.3761281967163086,
        "flow_mse": 2.304928779602051,
        "art_l2": 0.07119935005903244,
        "t_mean": 0.686140239238739,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09447285532951355,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12284307181835175,
        "stage2_mask_spatial_var_mean": 0.03497007489204407
      },
      {
        "step": 5,
        "loss": 3.4938910007476807,
        "flow_mse": 3.2933456897735596,
        "art_l2": 0.2005452662706375,
        "t_mean": 0.728419840335846,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.13586559891700745,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12117581814527512,
        "stage2_mask_spatial_var_mean": 0.06845814734697342
      },
      {
        "step": 6,
        "loss": 2.5839953422546387,
        "flow_mse": 2.5458407402038574,
        "art_l2": 0.03815459460020065,
        "t_mean": 0.6403265595436096,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 9.0,
        "stage2_mask_mass_mean": 0.03964406996965408,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.03179550915956497,
        "stage2_mask_spatial_var_mean": 0.1153540164232254
      },
      {
        "step": 7,
        "loss": 3.9821696281433105,
        "flow_mse": 3.8552777767181396,
        "art_l2": 0.1268918216228485,
        "t_mean": 0.5700072646141052,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10189920663833618,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17310911417007446,
        "stage2_mask_spatial_var_mean": 0.06178075820207596
      },
      {
        "step": 8,
        "loss": 2.688394546508789,
        "flow_mse": 2.5778989791870117,
        "art_l2": 0.11049561202526093,
        "t_mean": 0.655311644077301,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08826395124197006,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11497963964939117,
        "stage2_mask_spatial_var_mean": 0.030766749754548073
      },
      {
        "step": 9,
        "loss": 3.5982933044433594,
        "flow_mse": 3.4159693717956543,
        "art_l2": 0.1823238730430603,
        "t_mean": 0.6019466519355774,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.1391526758670807,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12259700894355774,
        "stage2_mask_spatial_var_mean": 0.06607501208782196
      },
      {
        "step": 10,
        "loss": 2.7572460174560547,
        "flow_mse": 2.583315372467041,
        "art_l2": 0.1739305853843689,
        "t_mean": 0.7785599827766418,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08728999644517899,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.05403045192360878,
        "stage2_mask_spatial_var_mean": 0.056121475994586945
      },
      {
        "step": 11,
        "loss": 4.719375133514404,
        "flow_mse": 4.687692642211914,
        "art_l2": 0.03168243542313576,
        "t_mean": 0.6584867835044861,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11102995276451111,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.22974403202533722,
        "stage2_mask_spatial_var_mean": 0.057314611971378326
      },
      {
        "step": 12,
        "loss": 2.25114107131958,
        "flow_mse": 2.1316981315612793,
        "art_l2": 0.11944302171468735,
        "t_mean": 0.2739311456680298,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13604821264743805,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1071624606847763,
        "stage2_mask_spatial_var_mean": 0.06341055035591125
      },
      {
        "step": 13,
        "loss": 3.6647603511810303,
        "flow_mse": 3.5567398071289062,
        "art_l2": 0.10802058130502701,
        "t_mean": 0.6072238683700562,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14481374621391296,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.2039666324853897,
        "stage2_mask_spatial_var_mean": 0.07484860718250275
      },
      {
        "step": 14,
        "loss": 1.8587486743927002,
        "flow_mse": 1.6864503622055054,
        "art_l2": 0.1722983568906784,
        "t_mean": 0.6395391821861267,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10025566816329956,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.08088136464357376,
        "stage2_mask_spatial_var_mean": 0.061420951038599014
      },
      {
        "step": 15,
        "loss": 1.9105596542358398,
        "flow_mse": 1.7941725254058838,
        "art_l2": 0.11638715863227844,
        "t_mean": 0.43723514676094055,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.1468224972486496,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10455434769392014,
        "stage2_mask_spatial_var_mean": 0.09443356096744537
      },
      {
        "step": 16,
        "loss": 2.5860650539398193,
        "flow_mse": 2.549393653869629,
        "art_l2": 0.03667141869664192,
        "t_mean": 0.6621525883674622,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 9.0,
        "stage2_mask_mass_mean": 0.03778471425175667,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10811017453670502,
        "stage2_mask_spatial_var_mean": 0.07324477285146713
      },
      {
        "step": 17,
        "loss": 2.233884334564209,
        "flow_mse": 2.162954807281494,
        "art_l2": 0.07092949002981186,
        "t_mean": 0.5240513682365417,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.07505478709936142,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1669202744960785,
        "stage2_mask_spatial_var_mean": 0.04181981086730957
      },
      {
        "step": 18,
        "loss": 1.9076623916625977,
        "flow_mse": 1.840387225151062,
        "art_l2": 0.06727514415979385,
        "t_mean": 0.5368333458900452,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 7.0,
        "stage2_mask_mass_mean": 0.05539323389530182,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17302262783050537,
        "stage2_mask_spatial_var_mean": 0.03860964998602867
      },
      {
        "step": 19,
        "loss": 2.7634294033050537,
        "flow_mse": 2.6960325241088867,
        "art_l2": 0.06739681959152222,
        "t_mean": 0.2161000519990921,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 8.0,
        "stage2_mask_mass_mean": 0.04516679048538208,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1213332787156105,
        "stage2_mask_spatial_var_mean": 0.04097684845328331
      },
      {
        "step": 20,
        "loss": 2.5328822135925293,
        "flow_mse": 2.4885265827178955,
        "art_l2": 0.04435553401708603,
        "t_mean": 0.5189581513404846,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13416120409965515,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1729474812746048,
        "stage2_mask_spatial_var_mean": 0.048083532601594925
      },
      {
        "step": 21,
        "loss": 5.953620433807373,
        "flow_mse": 5.783384799957275,
        "art_l2": 0.17023544013500214,
        "t_mean": 0.11379662156105042,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08619430661201477,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1525188833475113,
        "stage2_mask_spatial_var_mean": 0.03136561065912247
      },
      {
        "step": 22,
        "loss": 3.170708417892456,
        "flow_mse": 2.996443271636963,
        "art_l2": 0.17426517605781555,
        "t_mean": 0.4425402879714966,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08929875493049622,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.110529825091362,
        "stage2_mask_spatial_var_mean": 0.04687997326254845
      },
      {
        "step": 23,
        "loss": 5.712956428527832,
        "flow_mse": 5.571297645568848,
        "art_l2": 0.14165854454040527,
        "t_mean": 0.20127582550048828,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.11102995276451111,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13087493181228638,
        "stage2_mask_spatial_var_mean": 0.04991722106933594
      },
      {
        "step": 24,
        "loss": 4.49974250793457,
        "flow_mse": 4.380911350250244,
        "art_l2": 0.11883101612329483,
        "t_mean": 0.7069639563560486,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13111761212348938,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.14403989911079407,
        "stage2_mask_spatial_var_mean": 0.04339682310819626
      },
      {
        "step": 25,
        "loss": 8.878008842468262,
        "flow_mse": 8.747542381286621,
        "art_l2": 0.13046608865261078,
        "t_mean": 0.6600659489631653,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16581447422504425,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15605172514915466,
        "stage2_mask_spatial_var_mean": 0.060132116079330444
      },
      {
        "step": 26,
        "loss": 4.466785430908203,
        "flow_mse": 4.335524559020996,
        "art_l2": 0.13126100599765778,
        "t_mean": 0.4758807420730591,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10189919918775558,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13613399863243103,
        "stage2_mask_spatial_var_mean": 0.08625788986682892
      },
      {
        "step": 27,
        "loss": 2.072011709213257,
        "flow_mse": 1.9716483354568481,
        "art_l2": 0.10036339610815048,
        "t_mean": 0.9104493260383606,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.12326516211032867,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.09395356476306915,
        "stage2_mask_spatial_var_mean": 0.06172376871109009
      },
      {
        "step": 28,
        "loss": 1.0187994241714478,
        "flow_mse": 0.9332992434501648,
        "art_l2": 0.08550018072128296,
        "t_mean": 0.33697742223739624,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 3.0,
        "stage2_mask_mass_mean": 0.07137639820575714,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17038846015930176,
        "stage2_mask_spatial_var_mean": 0.035082150250673294
      },
      {
        "step": 29,
        "loss": 3.0617971420288086,
        "flow_mse": 2.9905381202697754,
        "art_l2": 0.07125909626483917,
        "t_mean": 0.324869304895401,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.0885683000087738,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.06510961055755615,
        "stage2_mask_spatial_var_mean": 0.0735899955034256
      },
      {
        "step": 30,
        "loss": 5.4423322677612305,
        "flow_mse": 5.302393913269043,
        "art_l2": 0.13993827998638153,
        "t_mean": 0.2462034672498703,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12016070634126663,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13704855740070343,
        "stage2_mask_spatial_var_mean": 0.03335332125425339
      },
      {
        "step": 31,
        "loss": 5.012214660644531,
        "flow_mse": 4.864922046661377,
        "art_l2": 0.1472925841808319,
        "t_mean": 0.3077169954776764,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13002191483974457,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1404089778661728,
        "stage2_mask_spatial_var_mean": 0.04379277676343918
      },
      {
        "step": 32,
        "loss": 3.4076669216156006,
        "flow_mse": 3.3593647480010986,
        "art_l2": 0.048302195966243744,
        "t_mean": 0.2582360506057739,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11443877220153809,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.18197765946388245,
        "stage2_mask_spatial_var_mean": 0.05140909552574158
      },
      {
        "step": 33,
        "loss": 2.0826330184936523,
        "flow_mse": 2.01558518409729,
        "art_l2": 0.06704793125391006,
        "t_mean": 0.5150696039199829,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 7.0,
        "stage2_mask_mass_mean": 0.04705381393432617,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16342011094093323,
        "stage2_mask_spatial_var_mean": 0.027939317747950554
      },
      {
        "step": 34,
        "loss": 2.758678913116455,
        "flow_mse": 2.591404676437378,
        "art_l2": 0.1672743558883667,
        "t_mean": 0.3112204372882843,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08132457733154297,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.19395869970321655,
        "stage2_mask_spatial_var_mean": 0.025912052020430565
      },
      {
        "step": 35,
        "loss": 3.6884732246398926,
        "flow_mse": 3.5606961250305176,
        "art_l2": 0.1277770698070526,
        "t_mean": 0.639766275882721,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12819576263427734,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13321955502033234,
        "stage2_mask_spatial_var_mean": 0.09279214590787888
      },
      {
        "step": 36,
        "loss": 2.638240337371826,
        "flow_mse": 2.594313144683838,
        "art_l2": 0.04392730072140694,
        "t_mean": 0.6604555249214172,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11200390756130219,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.19336432218551636,
        "stage2_mask_spatial_var_mean": 0.042206890881061554
      },
      {
        "step": 37,
        "loss": 2.148651599884033,
        "flow_mse": 1.9475352764129639,
        "art_l2": 0.20111630856990814,
        "t_mean": 0.756874680519104,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.11723886430263519,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.09051080793142319,
        "stage2_mask_spatial_var_mean": 0.06330938637256622
      },
      {
        "step": 38,
        "loss": 2.6695380210876465,
        "flow_mse": 2.580481767654419,
        "art_l2": 0.08905622363090515,
        "t_mean": 0.5363696813583374,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08162893354892731,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.06355810165405273,
        "stage2_mask_spatial_var_mean": 0.06627355515956879
      },
      {
        "step": 39,
        "loss": 2.4631924629211426,
        "flow_mse": 2.3919801712036133,
        "art_l2": 0.07121236622333527,
        "t_mean": 0.580784797668457,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12235209345817566,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.14516600966453552,
        "stage2_mask_spatial_var_mean": 0.08031938225030899
      },
      {
        "step": 40,
        "loss": 1.5618746280670166,
        "flow_mse": 1.4998279809951782,
        "art_l2": 0.06204666197299957,
        "t_mean": 0.38077911734580994,
        "stage2_mask_part_count": 10.0,
        "stage2_mask_empty_parts": 3.0,
        "stage2_mask_mass_mean": 0.05222791060805321,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12488780915737152,
        "stage2_mask_spatial_var_mean": 0.019491467624902725
      }
    ]
  },
  {
    "label": "Full Stage2 spatial adapter",
    "group": "stage2",
    "phase": "full",
    "note": "Full matrix Stage2 spatial adapter.",
    "rel_dir": "pact_ot_matrix_20260521_full_stage2_spatial_adapter",
    "status": "ok",
    "steps": 40,
    "loss": 5.98215389251709,
    "flow_mse": 5.825160503387451,
    "art_l2": 0.1569933295249939,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 6.0,
    "stage2_mask_empty_parts": 1.0,
    "stage2_mask_centroid_spread": 0.11533430963754654,
    "stage2_mask_spatial_var_mean": 0.046354711055755615,
    "stage2_spatial_adapter_l2": 2.1484917811813847e-20,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": true,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_20260521_full_stage2_spatial_adapter",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 40,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": true,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 7.412285804748535,
        "flow_mse": 7.278761386871338,
        "art_l2": 0.13352462649345398,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14974434673786163,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16794748604297638,
        "stage2_mask_spatial_var_mean": 0.04503370076417923,
        "stage2_spatial_adapter_l2": 0.0
      },
      {
        "step": 2,
        "loss": 3.420111656188965,
        "flow_mse": 3.3164515495300293,
        "art_l2": 0.10366011410951614,
        "t_mean": 0.3323965072631836,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.1470051258802414,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.21312789618968964,
        "stage2_mask_spatial_var_mean": 0.052936047315597534,
        "stage2_spatial_adapter_l2": 3.0241600085305823e-24
      },
      {
        "step": 3,
        "loss": 4.46923303604126,
        "flow_mse": 4.312280178070068,
        "art_l2": 0.15695306658744812,
        "t_mean": 0.8121371269226074,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09739469736814499,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15333124995231628,
        "stage2_mask_spatial_var_mean": 0.054026901721954346,
        "stage2_spatial_adapter_l2": 1.5990617601057003e-23
      },
      {
        "step": 4,
        "loss": 7.818053245544434,
        "flow_mse": 7.688683986663818,
        "art_l2": 0.12936925888061523,
        "t_mean": 0.686140239238739,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14919649064540863,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17030486464500427,
        "stage2_mask_spatial_var_mean": 0.0551006905734539,
        "stage2_spatial_adapter_l2": 1.3048428326403163e-22
      },
      {
        "step": 5,
        "loss": 3.3667635917663574,
        "flow_mse": 3.252732276916504,
        "art_l2": 0.11403129994869232,
        "t_mean": 0.728419840335846,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08144631236791611,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12281860411167145,
        "stage2_mask_spatial_var_mean": 0.0692753717303276,
        "stage2_spatial_adapter_l2": 1.7243811501421965e-22
      },
      {
        "step": 6,
        "loss": 2.29494047164917,
        "flow_mse": 2.200533866882324,
        "art_l2": 0.09440654516220093,
        "t_mean": 0.6403265595436096,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.044823698699474335,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10919459164142609,
        "stage2_mask_spatial_var_mean": 0.015086439438164234,
        "stage2_spatial_adapter_l2": 1.6615630834086163e-22
      },
      {
        "step": 7,
        "loss": 1.9948036670684814,
        "flow_mse": 1.9262633323669434,
        "art_l2": 0.0685403123497963,
        "t_mean": 0.5700072646141052,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 6.0,
        "stage2_mask_mass_mean": 0.04541027545928955,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.0890435129404068,
        "stage2_mask_spatial_var_mean": 0.02842908911406994,
        "stage2_spatial_adapter_l2": 1.5779074767446764e-22
      },
      {
        "step": 8,
        "loss": 2.739849805831909,
        "flow_mse": 2.6751482486724854,
        "art_l2": 0.06470150500535965,
        "t_mean": 0.655311644077301,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09660336375236511,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1617322862148285,
        "stage2_mask_spatial_var_mean": 0.061530254781246185,
        "stage2_spatial_adapter_l2": 2.8591381247699174e-22
      },
      {
        "step": 9,
        "loss": 8.744171142578125,
        "flow_mse": 8.615007400512695,
        "art_l2": 0.12916402518749237,
        "t_mean": 0.6019466519355774,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16745799779891968,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11526110768318176,
        "stage2_mask_spatial_var_mean": 0.06515448540449142,
        "stage2_spatial_adapter_l2": 3.610239981881472e-22
      },
      {
        "step": 10,
        "loss": 1.8667640686035156,
        "flow_mse": 1.8235417604446411,
        "art_l2": 0.04322226345539093,
        "t_mean": 0.7785599827766418,
        "stage2_mask_part_count": 10.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.05018261820077896,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.135658860206604,
        "stage2_mask_spatial_var_mean": 0.018480516970157623,
        "stage2_spatial_adapter_l2": 3.6648286516517867e-22
      },
      {
        "step": 11,
        "loss": 3.2716774940490723,
        "flow_mse": 3.0789954662323,
        "art_l2": 0.19268211722373962,
        "t_mean": 0.6584867835044861,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.11102995276451111,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13087493181228638,
        "stage2_mask_spatial_var_mean": 0.04991722106933594,
        "stage2_spatial_adapter_l2": 6.463716231053565e-22
      },
      {
        "step": 12,
        "loss": 3.0916881561279297,
        "flow_mse": 3.0212321281433105,
        "art_l2": 0.07045593112707138,
        "t_mean": 0.2739311456680298,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09934259206056595,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13076920807361603,
        "stage2_mask_spatial_var_mean": 0.07914616167545319,
        "stage2_spatial_adapter_l2": 9.84228303741875e-22
      },
      {
        "step": 13,
        "loss": 3.9065678119659424,
        "flow_mse": 3.789628267288208,
        "art_l2": 0.11693961173295975,
        "t_mean": 0.6072238683700562,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08315072953701019,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1498417854309082,
        "stage2_mask_spatial_var_mean": 0.03243282437324524,
        "stage2_spatial_adapter_l2": 1.0534653001068025e-21
      },
      {
        "step": 14,
        "loss": 1.8581064939498901,
        "flow_mse": 1.684683918952942,
        "art_l2": 0.17342253029346466,
        "t_mean": 0.6395391821861267,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.06902848929166794,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.05083201825618744,
        "stage2_mask_spatial_var_mean": 0.05934283882379532,
        "stage2_spatial_adapter_l2": 1.2205194331320015e-21
      },
      {
        "step": 15,
        "loss": 3.0207626819610596,
        "flow_mse": 2.852099657058716,
        "art_l2": 0.16866302490234375,
        "t_mean": 0.43723514676094055,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10390797257423401,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11111626029014587,
        "stage2_mask_spatial_var_mean": 0.05304690822958946,
        "stage2_spatial_adapter_l2": 2.165921744923777e-21
      },
      {
        "step": 16,
        "loss": 3.889638662338257,
        "flow_mse": 3.7034218311309814,
        "art_l2": 0.18621674180030823,
        "t_mean": 0.6621525883674622,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12016070634126663,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13704855740070343,
        "stage2_mask_spatial_var_mean": 0.03335332125425339,
        "stage2_spatial_adapter_l2": 2.5698326647843567e-21
      },
      {
        "step": 17,
        "loss": 2.4713966846466064,
        "flow_mse": 2.4245176315307617,
        "art_l2": 0.046879060566425323,
        "t_mean": 0.5240513682365417,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12271732091903687,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15761402249336243,
        "stage2_mask_spatial_var_mean": 0.05832899361848831,
        "stage2_spatial_adapter_l2": 3.322792194664398e-21
      },
      {
        "step": 18,
        "loss": 7.127469062805176,
        "flow_mse": 6.998048305511475,
        "art_l2": 0.12942074239253998,
        "t_mean": 0.5368333458900452,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.036036036908626556,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16098950803279877,
        "stage2_mask_spatial_var_mean": 0.032539673149585724,
        "stage2_spatial_adapter_l2": 3.5280319412895e-21
      },
      {
        "step": 19,
        "loss": 4.7238287925720215,
        "flow_mse": 4.6226806640625,
        "art_l2": 0.10114829987287521,
        "t_mean": 0.2161000519990921,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10810810327529907,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.18285012245178223,
        "stage2_mask_spatial_var_mean": 0.0355285182595253,
        "stage2_spatial_adapter_l2": 3.1351007535392464e-21
      },
      {
        "step": 20,
        "loss": 7.625058174133301,
        "flow_mse": 7.5084147453308105,
        "art_l2": 0.11664321273565292,
        "t_mean": 0.5189581513404846,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11127345263957977,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1711958348751068,
        "stage2_mask_spatial_var_mean": 0.03285317122936249,
        "stage2_spatial_adapter_l2": 3.807792663949317e-21
      },
      {
        "step": 21,
        "loss": 1.8952394723892212,
        "flow_mse": 1.82036554813385,
        "art_l2": 0.07487387210130692,
        "t_mean": 0.11379662156105042,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.07638527452945709,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1542912721633911,
        "stage2_mask_spatial_var_mean": 0.02149767056107521,
        "stage2_spatial_adapter_l2": 3.369800933499272e-21
      },
      {
        "step": 22,
        "loss": 3.218903064727783,
        "flow_mse": 3.039728879928589,
        "art_l2": 0.17917414009571075,
        "t_mean": 0.4425402879714966,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.15157049894332886,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12493190169334412,
        "stage2_mask_spatial_var_mean": 0.07618506997823715,
        "stage2_spatial_adapter_l2": 3.638611610462348e-21
      },
      {
        "step": 23,
        "loss": 4.548165321350098,
        "flow_mse": 4.420348644256592,
        "art_l2": 0.12781670689582825,
        "t_mean": 0.20127582550048828,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08777697384357452,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15772874653339386,
        "stage2_mask_spatial_var_mean": 0.0357699990272522,
        "stage2_spatial_adapter_l2": 4.2459039478472695e-21
      },
      {
        "step": 24,
        "loss": 1.9257347583770752,
        "flow_mse": 1.739397644996643,
        "art_l2": 0.1863371580839157,
        "t_mean": 0.7069639563560486,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.07651570439338684,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.06106841191649437,
        "stage2_mask_spatial_var_mean": 0.05943670868873596,
        "stage2_spatial_adapter_l2": 5.161581980730317e-21
      },
      {
        "step": 25,
        "loss": 8.639487266540527,
        "flow_mse": 8.508219718933105,
        "art_l2": 0.13126729428768158,
        "t_mean": 0.6600659489631653,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16581445932388306,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15902292728424072,
        "stage2_mask_spatial_var_mean": 0.06032077595591545,
        "stage2_spatial_adapter_l2": 7.06985372656034e-21
      },
      {
        "step": 26,
        "loss": 6.612944602966309,
        "flow_mse": 6.570949554443359,
        "art_l2": 0.04199489578604698,
        "t_mean": 0.4758807420730591,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.18504993617534637,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15875029563903809,
        "stage2_mask_spatial_var_mean": 0.05624683201313019,
        "stage2_spatial_adapter_l2": 8.180788793168168e-21
      },
      {
        "step": 27,
        "loss": 3.420633554458618,
        "flow_mse": 3.2958340644836426,
        "art_l2": 0.12479950487613678,
        "t_mean": 0.9104493260383606,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09252496808767319,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.142442524433136,
        "stage2_mask_spatial_var_mean": 0.02995327115058899,
        "stage2_spatial_adapter_l2": 7.409329780863097e-21
      },
      {
        "step": 28,
        "loss": 4.488325595855713,
        "flow_mse": 4.336333274841309,
        "art_l2": 0.1519920974969864,
        "t_mean": 0.33697742223739624,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.15029218792915344,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.210830956697464,
        "stage2_mask_spatial_var_mean": 0.05848394334316254,
        "stage2_spatial_adapter_l2": 9.028949420241925e-21
      },
      {
        "step": 29,
        "loss": 1.0367701053619385,
        "flow_mse": 0.954574465751648,
        "art_l2": 0.08219566941261292,
        "t_mean": 0.324869304895401,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 3.0,
        "stage2_mask_mass_mean": 0.07137639820575714,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17038846015930176,
        "stage2_mask_spatial_var_mean": 0.035082150250673294,
        "stage2_spatial_adapter_l2": 1.0091704447342806e-20
      },
      {
        "step": 30,
        "loss": 3.3687708377838135,
        "flow_mse": 3.3064093589782715,
        "art_l2": 0.06236148253083229,
        "t_mean": 0.2462034672498703,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12198685109615326,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10257704555988312,
        "stage2_mask_spatial_var_mean": 0.06817074865102768,
        "stage2_spatial_adapter_l2": 1.0567791318513255e-20
      },
      {
        "step": 31,
        "loss": 3.1523396968841553,
        "flow_mse": 3.088102102279663,
        "art_l2": 0.06423763185739517,
        "t_mean": 0.3077169954776764,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.09495982527732849,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.0754053071141243,
        "stage2_mask_spatial_var_mean": 0.07356518507003784,
        "stage2_spatial_adapter_l2": 1.229500179086723e-20
      },
      {
        "step": 32,
        "loss": 3.1845195293426514,
        "flow_mse": 3.0719213485717773,
        "art_l2": 0.11259808391332626,
        "t_mean": 0.2582360506057739,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.05717511847615242,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.14053958654403687,
        "stage2_mask_spatial_var_mean": 0.02332073263823986,
        "stage2_spatial_adapter_l2": 1.0436328570047711e-20
      },
      {
        "step": 33,
        "loss": 2.939969301223755,
        "flow_mse": 2.7102608680725098,
        "art_l2": 0.22970837354660034,
        "t_mean": 0.5150696039199829,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12417824566364288,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10249984264373779,
        "stage2_mask_spatial_var_mean": 0.06571410596370697,
        "stage2_spatial_adapter_l2": 1.1873898196624549e-20
      },
      {
        "step": 34,
        "loss": 5.846395015716553,
        "flow_mse": 5.694240093231201,
        "art_l2": 0.1521550863981247,
        "t_mean": 0.3112204372882843,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10153397172689438,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12184513360261917,
        "stage2_mask_spatial_var_mean": 0.06188197433948517,
        "stage2_spatial_adapter_l2": 1.3430284563376386e-20
      },
      {
        "step": 35,
        "loss": 3.002622365951538,
        "flow_mse": 2.802635669708252,
        "art_l2": 0.19998668134212494,
        "t_mean": 0.639766275882721,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16995374858379364,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.09016727656126022,
        "stage2_mask_spatial_var_mean": 0.07749043405056,
        "stage2_spatial_adapter_l2": 1.4286511009216103e-20
      },
      {
        "step": 36,
        "loss": 2.8920681476593018,
        "flow_mse": 2.7428979873657227,
        "art_l2": 0.1491701602935791,
        "t_mean": 0.6604555249214172,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10007305443286896,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13893088698387146,
        "stage2_mask_spatial_var_mean": 0.03599216043949127,
        "stage2_spatial_adapter_l2": 1.7027668005768928e-20
      },
      {
        "step": 37,
        "loss": 1.7985066175460815,
        "flow_mse": 1.6943010091781616,
        "art_l2": 0.10420558601617813,
        "t_mean": 0.756874680519104,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.1468224972486496,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10455434769392014,
        "stage2_mask_spatial_var_mean": 0.09443356096744537,
        "stage2_spatial_adapter_l2": 2.0168052738977962e-20
      },
      {
        "step": 38,
        "loss": 2.981696367263794,
        "flow_mse": 2.8329010009765625,
        "art_l2": 0.14879527688026428,
        "t_mean": 0.5363696813583374,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11541271209716797,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10026109218597412,
        "stage2_mask_spatial_var_mean": 0.04383675381541252,
        "stage2_spatial_adapter_l2": 2.1629141275256442e-20
      },
      {
        "step": 39,
        "loss": 2.542611598968506,
        "flow_mse": 2.470895767211914,
        "art_l2": 0.0717158392071724,
        "t_mean": 0.580784797668457,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09435111284255981,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1347745954990387,
        "stage2_mask_spatial_var_mean": 0.030499106273055077,
        "stage2_spatial_adapter_l2": 2.0883673828761436e-20
      },
      {
        "step": 40,
        "loss": 5.98215389251709,
        "flow_mse": 5.825160503387451,
        "art_l2": 0.1569933295249939,
        "t_mean": 0.38077911734580994,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10311663150787354,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11533430963754654,
        "stage2_mask_spatial_var_mean": 0.046354711055755615,
        "stage2_spatial_adapter_l2": 2.1484917811813847e-20
      }
    ]
  },
  {
    "label": "Official full Stage1 baseline",
    "group": "stage1",
    "phase": "official-full",
    "note": "Official-initialized full matrix baseline with final checkpoint export.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage1_baseline",
    "status": "ok",
    "steps": 60,
    "loss": 0.04794942960143089,
    "flow_mse": 0.04794942960143089,
    "art_l2": null,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage1_baseline",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.03042520396411419,
        "flow_mse": 0.03042520396411419,
        "t_mean": 0.4452204704284668
      },
      {
        "step": 2,
        "loss": 0.062441423535346985,
        "flow_mse": 0.062441423535346985,
        "t_mean": 0.25286394357681274
      },
      {
        "step": 3,
        "loss": 0.0608043372631073,
        "flow_mse": 0.0608043372631073,
        "t_mean": 0.6159385442733765
      },
      {
        "step": 4,
        "loss": 0.04127635434269905,
        "flow_mse": 0.04127635434269905,
        "t_mean": 0.6475216150283813
      },
      {
        "step": 5,
        "loss": 0.050262775272130966,
        "flow_mse": 0.050262775272130966,
        "t_mean": 0.6798597574234009
      },
      {
        "step": 6,
        "loss": 0.051486220210790634,
        "flow_mse": 0.051486220210790634,
        "t_mean": 0.5003219842910767
      },
      {
        "step": 7,
        "loss": 0.05220533534884453,
        "flow_mse": 0.05220533534884453,
        "t_mean": 0.6674641966819763
      },
      {
        "step": 8,
        "loss": 0.10473670810461044,
        "flow_mse": 0.10473670810461044,
        "t_mean": 0.6387549638748169
      },
      {
        "step": 9,
        "loss": 0.03512171655893326,
        "flow_mse": 0.03512171655893326,
        "t_mean": 0.5298079252243042
      },
      {
        "step": 10,
        "loss": 0.03571426868438721,
        "flow_mse": 0.03571426868438721,
        "t_mean": 0.5381855368614197
      },
      {
        "step": 11,
        "loss": 0.10064953565597534,
        "flow_mse": 0.10064953565597534,
        "t_mean": 0.5974305868148804
      },
      {
        "step": 12,
        "loss": 0.043222926557064056,
        "flow_mse": 0.043222926557064056,
        "t_mean": 0.4660314917564392
      },
      {
        "step": 13,
        "loss": 0.02778242528438568,
        "flow_mse": 0.02778242528438568,
        "t_mean": 0.42932528257369995
      },
      {
        "step": 14,
        "loss": 0.06667459011077881,
        "flow_mse": 0.06667459011077881,
        "t_mean": 0.44856566190719604
      },
      {
        "step": 15,
        "loss": 0.04190537706017494,
        "flow_mse": 0.04190537706017494,
        "t_mean": 0.34305432438850403
      },
      {
        "step": 16,
        "loss": 0.03869957476854324,
        "flow_mse": 0.03869957476854324,
        "t_mean": 0.4998842179775238
      },
      {
        "step": 17,
        "loss": 0.038947656750679016,
        "flow_mse": 0.038947656750679016,
        "t_mean": 0.4249878525733948
      },
      {
        "step": 18,
        "loss": 0.027833590283989906,
        "flow_mse": 0.027833590283989906,
        "t_mean": 0.42435136437416077
      },
      {
        "step": 19,
        "loss": 0.06748953461647034,
        "flow_mse": 0.06748953461647034,
        "t_mean": 0.49784499406814575
      },
      {
        "step": 20,
        "loss": 0.049178969115018845,
        "flow_mse": 0.049178969115018845,
        "t_mean": 0.4546642005443573
      },
      {
        "step": 21,
        "loss": 0.1117495745420456,
        "flow_mse": 0.1117495745420456,
        "t_mean": 0.31838786602020264
      },
      {
        "step": 22,
        "loss": 0.09413465857505798,
        "flow_mse": 0.09413465857505798,
        "t_mean": 0.5307409167289734
      },
      {
        "step": 23,
        "loss": 0.10635565966367722,
        "flow_mse": 0.10635565966367722,
        "t_mean": 0.511449933052063
      },
      {
        "step": 24,
        "loss": 0.05882437154650688,
        "flow_mse": 0.05882437154650688,
        "t_mean": 0.5524937510490417
      },
      {
        "step": 25,
        "loss": 0.05057203024625778,
        "flow_mse": 0.05057203024625778,
        "t_mean": 0.4828733801841736
      },
      {
        "step": 26,
        "loss": 0.038350582122802734,
        "flow_mse": 0.038350582122802734,
        "t_mean": 0.47976693511009216
      },
      {
        "step": 27,
        "loss": 0.025239141657948494,
        "flow_mse": 0.025239141657948494,
        "t_mean": 0.7675442695617676
      },
      {
        "step": 28,
        "loss": 0.15928640961647034,
        "flow_mse": 0.15928640961647034,
        "t_mean": 0.6012675166130066
      },
      {
        "step": 29,
        "loss": 0.02960882894694805,
        "flow_mse": 0.02960882894694805,
        "t_mean": 0.6016367673873901
      },
      {
        "step": 30,
        "loss": 0.05642203986644745,
        "flow_mse": 0.05642203986644745,
        "t_mean": 0.5350275039672852
      },
      {
        "step": 31,
        "loss": 0.04145316034555435,
        "flow_mse": 0.04145316034555435,
        "t_mean": 0.3996628522872925
      },
      {
        "step": 32,
        "loss": 0.049057237803936005,
        "flow_mse": 0.049057237803936005,
        "t_mean": 0.4242017865180969
      },
      {
        "step": 33,
        "loss": 0.03532620891928673,
        "flow_mse": 0.03532620891928673,
        "t_mean": 0.6198998689651489
      },
      {
        "step": 34,
        "loss": 0.037098564207553864,
        "flow_mse": 0.037098564207553864,
        "t_mean": 0.4256276786327362
      },
      {
        "step": 35,
        "loss": 0.04101256653666496,
        "flow_mse": 0.04101256653666496,
        "t_mean": 0.5468468070030212
      },
      {
        "step": 36,
        "loss": 0.09452317655086517,
        "flow_mse": 0.09452317655086517,
        "t_mean": 0.4543689489364624
      },
      {
        "step": 37,
        "loss": 0.14154499769210815,
        "flow_mse": 0.14154499769210815,
        "t_mean": 0.5334687829017639
      },
      {
        "step": 38,
        "loss": 0.0858859047293663,
        "flow_mse": 0.0858859047293663,
        "t_mean": 0.4061962068080902
      },
      {
        "step": 39,
        "loss": 0.04948152229189873,
        "flow_mse": 0.04948152229189873,
        "t_mean": 0.6304954886436462
      },
      {
        "step": 40,
        "loss": 0.03951293230056763,
        "flow_mse": 0.03951293230056763,
        "t_mean": 0.4235572814941406
      },
      {
        "step": 41,
        "loss": 0.08911201357841492,
        "flow_mse": 0.08911201357841492,
        "t_mean": 0.3948693871498108
      },
      {
        "step": 42,
        "loss": 0.199324369430542,
        "flow_mse": 0.199324369430542,
        "t_mean": 0.5838073492050171
      },
      {
        "step": 43,
        "loss": 0.05309363454580307,
        "flow_mse": 0.05309363454580307,
        "t_mean": 0.674645185470581
      },
      {
        "step": 44,
        "loss": 0.05154676362872124,
        "flow_mse": 0.05154676362872124,
        "t_mean": 0.4860682487487793
      },
      {
        "step": 45,
        "loss": 0.02955843135714531,
        "flow_mse": 0.02955843135714531,
        "t_mean": 0.3828374445438385
      },
      {
        "step": 46,
        "loss": 0.06477203220129013,
        "flow_mse": 0.06477203220129013,
        "t_mean": 0.3058706820011139
      },
      {
        "step": 47,
        "loss": 0.04948100075125694,
        "flow_mse": 0.04948100075125694,
        "t_mean": 0.31310853362083435
      },
      {
        "step": 48,
        "loss": 0.07191435992717743,
        "flow_mse": 0.07191435992717743,
        "t_mean": 0.5626304149627686
      },
      {
        "step": 49,
        "loss": 0.0598481222987175,
        "flow_mse": 0.0598481222987175,
        "t_mean": 0.7366573214530945
      },
      {
        "step": 50,
        "loss": 0.04775416851043701,
        "flow_mse": 0.04775416851043701,
        "t_mean": 0.5806166529655457
      },
      {
        "step": 51,
        "loss": 0.054236020892858505,
        "flow_mse": 0.054236020892858505,
        "t_mean": 0.6815006732940674
      },
      {
        "step": 52,
        "loss": 0.03817752003669739,
        "flow_mse": 0.03817752003669739,
        "t_mean": 0.509903609752655
      },
      {
        "step": 53,
        "loss": 0.023379569873213768,
        "flow_mse": 0.023379569873213768,
        "t_mean": 0.5933268070220947
      },
      {
        "step": 54,
        "loss": 0.1525423228740692,
        "flow_mse": 0.1525423228740692,
        "t_mean": 0.5004596710205078
      },
      {
        "step": 55,
        "loss": 0.11484219878911972,
        "flow_mse": 0.11484219878911972,
        "t_mean": 0.7171943187713623
      },
      {
        "step": 56,
        "loss": 0.08302053809165955,
        "flow_mse": 0.08302053809165955,
        "t_mean": 0.4547177255153656
      },
      {
        "step": 57,
        "loss": 0.039569154381752014,
        "flow_mse": 0.039569154381752014,
        "t_mean": 0.43234071135520935
      },
      {
        "step": 58,
        "loss": 0.037049632519483566,
        "flow_mse": 0.037049632519483566,
        "t_mean": 0.5147815942764282
      },
      {
        "step": 59,
        "loss": 0.09980382770299911,
        "flow_mse": 0.09980382770299911,
        "t_mean": 0.49480095505714417
      },
      {
        "step": 60,
        "loss": 0.04794942960143089,
        "flow_mse": 0.04794942960143089,
        "t_mean": 0.4051949977874756
      }
    ]
  },
  {
    "label": "Official full Stage1 OT first",
    "group": "stage1",
    "phase": "official-full",
    "note": "Official-initialized full matrix with one OT-gated cross-attention block.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage1_ot_first",
    "status": "ok",
    "steps": 60,
    "loss": 0.04790760949254036,
    "flow_mse": 0.04790760949254036,
    "art_l2": null,
    "ot_gate_mean": 0.93505859375,
    "ot_edge_delta": 0.0,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage1_ot_first",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.030395058915019035,
        "flow_mse": 0.030395058915019035,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.98095703125,
        "ot_gate_min": 0.94921875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 0.062428127974271774,
        "flow_mse": 0.062428127974271774,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.98095703125,
        "ot_gate_min": 0.9482421875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 0.06080120801925659,
        "flow_mse": 0.06080120801925659,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.94873046875,
        "ot_gate_min": 0.89892578125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 0.04123881086707115,
        "flow_mse": 0.04123881086707115,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96728515625,
        "ot_gate_min": 0.9140625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 0.05021090805530548,
        "flow_mse": 0.05021090805530548,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.95703125,
        "ot_gate_min": 0.88134765625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 0.05141184851527214,
        "flow_mse": 0.05141184851527214,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9462890625,
        "ot_gate_min": 0.83984375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 0.052698563784360886,
        "flow_mse": 0.052698563784360886,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.92626953125,
        "ot_gate_min": 0.79931640625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 0.10470953583717346,
        "flow_mse": 0.10470953583717346,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.99169921875,
        "ot_gate_min": 0.982421875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 0.03509858250617981,
        "flow_mse": 0.03509858250617981,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9755859375,
        "ot_gate_min": 0.93896484375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 0.03565007448196411,
        "flow_mse": 0.03565007448196411,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.95703125,
        "ot_gate_min": 0.876953125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 0.10058961063623428,
        "flow_mse": 0.10058961063623428,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9833984375,
        "ot_gate_min": 0.966796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 0.0432087741792202,
        "flow_mse": 0.0432087741792202,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96875,
        "ot_gate_min": 0.9130859375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 0.027736477553844452,
        "flow_mse": 0.027736477553844452,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.95703125,
        "ot_gate_min": 0.9150390625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 0.06664015352725983,
        "flow_mse": 0.06664015352725983,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96337890625,
        "ot_gate_min": 0.8935546875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 0.04186258465051651,
        "flow_mse": 0.04186258465051651,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9697265625,
        "ot_gate_min": 0.9150390625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 0.03865876793861389,
        "flow_mse": 0.03865876793861389,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9638671875,
        "ot_gate_min": 0.88037109375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 0.03892654925584793,
        "flow_mse": 0.03892654925584793,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9755859375,
        "ot_gate_min": 0.9287109375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 0.027803665027022362,
        "flow_mse": 0.027803665027022362,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9833984375,
        "ot_gate_min": 0.935546875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 0.06736179441213608,
        "flow_mse": 0.06736179441213608,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9326171875,
        "ot_gate_min": 0.84423828125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 0.04913199692964554,
        "flow_mse": 0.04913199692964554,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.970703125,
        "ot_gate_min": 0.91064453125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 21,
        "loss": 0.1117626279592514,
        "flow_mse": 0.1117626279592514,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9658203125,
        "ot_gate_min": 0.90576171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 22,
        "loss": 0.09411247819662094,
        "flow_mse": 0.09411247819662094,
        "t_mean": 0.5307409167289734,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9736328125,
        "ot_gate_min": 0.94677734375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 23,
        "loss": 0.1063685342669487,
        "flow_mse": 0.1063685342669487,
        "t_mean": 0.511449933052063,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.92724609375,
        "ot_gate_min": 0.85595703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 24,
        "loss": 0.05877227708697319,
        "flow_mse": 0.05877227708697319,
        "t_mean": 0.5524937510490417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9423828125,
        "ot_gate_min": 0.79736328125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 25,
        "loss": 0.05054879188537598,
        "flow_mse": 0.05054879188537598,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96875,
        "ot_gate_min": 0.9130859375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 26,
        "loss": 0.03832431137561798,
        "flow_mse": 0.03832431137561798,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96337890625,
        "ot_gate_min": 0.900390625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 27,
        "loss": 0.025102460756897926,
        "flow_mse": 0.025102460756897926,
        "t_mean": 0.7675442695617676,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.93359375,
        "ot_gate_min": 0.86962890625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 28,
        "loss": 0.15925538539886475,
        "flow_mse": 0.15925538539886475,
        "t_mean": 0.6012675166130066,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.95556640625,
        "ot_gate_min": 0.876953125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 29,
        "loss": 0.02956259809434414,
        "flow_mse": 0.02956259809434414,
        "t_mean": 0.6016367673873901,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9638671875,
        "ot_gate_min": 0.8701171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 30,
        "loss": 0.056098517030477524,
        "flow_mse": 0.056098517030477524,
        "t_mean": 0.5350275039672852,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.94091796875,
        "ot_gate_min": 0.830078125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 31,
        "loss": 0.04138873517513275,
        "flow_mse": 0.04138873517513275,
        "t_mean": 0.3996628522872925,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97119140625,
        "ot_gate_min": 0.91943359375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 32,
        "loss": 0.04903041198849678,
        "flow_mse": 0.04903041198849678,
        "t_mean": 0.4242017865180969,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9453125,
        "ot_gate_min": 0.76708984375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 33,
        "loss": 0.035293783992528915,
        "flow_mse": 0.035293783992528915,
        "t_mean": 0.6198998689651489,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.974609375,
        "ot_gate_min": 0.9345703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 34,
        "loss": 0.037074971944093704,
        "flow_mse": 0.037074971944093704,
        "t_mean": 0.4256276786327362,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9697265625,
        "ot_gate_min": 0.91650390625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 35,
        "loss": 0.04097188636660576,
        "flow_mse": 0.04097188636660576,
        "t_mean": 0.5468468070030212,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96728515625,
        "ot_gate_min": 0.9013671875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 36,
        "loss": 0.09467152506113052,
        "flow_mse": 0.09467152506113052,
        "t_mean": 0.4543689489364624,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.95458984375,
        "ot_gate_min": 0.84326171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 37,
        "loss": 0.14168372750282288,
        "flow_mse": 0.14168372750282288,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9501953125,
        "ot_gate_min": 0.85400390625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 38,
        "loss": 0.08585257828235626,
        "flow_mse": 0.08585257828235626,
        "t_mean": 0.4061962068080902,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9462890625,
        "ot_gate_min": 0.85595703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 39,
        "loss": 0.049457620829343796,
        "flow_mse": 0.049457620829343796,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.98095703125,
        "ot_gate_min": 0.94677734375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 40,
        "loss": 0.0394941009581089,
        "flow_mse": 0.0394941009581089,
        "t_mean": 0.4235572814941406,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9873046875,
        "ot_gate_min": 0.97509765625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 41,
        "loss": 0.08924275636672974,
        "flow_mse": 0.08924275636672974,
        "t_mean": 0.3948693871498108,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.94580078125,
        "ot_gate_min": 0.8916015625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 42,
        "loss": 0.19933535158634186,
        "flow_mse": 0.19933535158634186,
        "t_mean": 0.5838073492050171,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.95703125,
        "ot_gate_min": 0.892578125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 43,
        "loss": 0.05304228514432907,
        "flow_mse": 0.05304228514432907,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96435546875,
        "ot_gate_min": 0.900390625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 44,
        "loss": 0.051508091390132904,
        "flow_mse": 0.051508091390132904,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96826171875,
        "ot_gate_min": 0.91064453125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 45,
        "loss": 0.0295268464833498,
        "flow_mse": 0.0295268464833498,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.962890625,
        "ot_gate_min": 0.927734375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 46,
        "loss": 0.06477928161621094,
        "flow_mse": 0.06477928161621094,
        "t_mean": 0.3058706820011139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97412109375,
        "ot_gate_min": 0.9130859375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 47,
        "loss": 0.04953252151608467,
        "flow_mse": 0.04953252151608467,
        "t_mean": 0.31310853362083435,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9697265625,
        "ot_gate_min": 0.8916015625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 48,
        "loss": 0.07188794016838074,
        "flow_mse": 0.07188794016838074,
        "t_mean": 0.5626304149627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9638671875,
        "ot_gate_min": 0.89697265625,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 49,
        "loss": 0.05987376719713211,
        "flow_mse": 0.05987376719713211,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.98291015625,
        "ot_gate_min": 0.95654296875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 50,
        "loss": 0.04773484915494919,
        "flow_mse": 0.04773484915494919,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.98583984375,
        "ot_gate_min": 0.95751953125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 51,
        "loss": 0.05399956926703453,
        "flow_mse": 0.05399956926703453,
        "t_mean": 0.6815006732940674,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.94970703125,
        "ot_gate_min": 0.859375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 52,
        "loss": 0.03819747269153595,
        "flow_mse": 0.03819747269153595,
        "t_mean": 0.509903609752655,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.97900390625,
        "ot_gate_min": 0.93896484375,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 53,
        "loss": 0.023327216506004333,
        "flow_mse": 0.023327216506004333,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96826171875,
        "ot_gate_min": 0.908203125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 54,
        "loss": 0.15251091122627258,
        "flow_mse": 0.15251091122627258,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.958984375,
        "ot_gate_min": 0.892578125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 55,
        "loss": 0.11540129780769348,
        "flow_mse": 0.11540129780769348,
        "t_mean": 0.7171943187713623,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.93408203125,
        "ot_gate_min": 0.814453125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 56,
        "loss": 0.08300577104091644,
        "flow_mse": 0.08300577104091644,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9619140625,
        "ot_gate_min": 0.8935546875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 57,
        "loss": 0.03953910246491432,
        "flow_mse": 0.03953910246491432,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.9765625,
        "ot_gate_min": 0.93701171875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 58,
        "loss": 0.03701750934123993,
        "flow_mse": 0.03701750934123993,
        "t_mean": 0.5147815942764282,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.96630859375,
        "ot_gate_min": 0.90966796875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 59,
        "loss": 0.09969684481620789,
        "flow_mse": 0.09969684481620789,
        "t_mean": 0.49480095505714417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.966796875,
        "ot_gate_min": 0.93310546875,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 60,
        "loss": 0.04790760949254036,
        "flow_mse": 0.04790760949254036,
        "t_mean": 0.4051949977874756,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.93505859375,
        "ot_gate_min": 0.703125,
        "ot_mask_prior_beta": 0.0,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Official full Stage1 mask prior",
    "group": "stage1",
    "phase": "official-full",
    "note": "Official-initialized full matrix with mask labels as a soft OT prior.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage1_mask_prior",
    "status": "ok",
    "steps": 60,
    "loss": 0.04789481684565544,
    "flow_mse": 0.04789481684565544,
    "art_l2": null,
    "ot_gate_mean": 0.268310546875,
    "ot_edge_delta": 0.0,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage1_mask_prior",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.02921455167233944,
        "flow_mse": 0.02921455167233944,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29248046875,
        "ot_gate_min": 0.0208740234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 0.06307478994131088,
        "flow_mse": 0.06307478994131088,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1976318359375,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 0.060768574476242065,
        "flow_mse": 0.060768574476242065,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3642578125,
        "ot_gate_min": 0.022430419921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 0.04302801936864853,
        "flow_mse": 0.04302801936864853,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26806640625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 0.04972422122955322,
        "flow_mse": 0.04972422122955322,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2305908203125,
        "ot_gate_min": 0.0218353271484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 0.05110488086938858,
        "flow_mse": 0.05110488086938858,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2308349609375,
        "ot_gate_min": 0.0216217041015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 0.0550682470202446,
        "flow_mse": 0.0550682470202446,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.329345703125,
        "ot_gate_min": 0.0217132568359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 0.10009296238422394,
        "flow_mse": 0.10009296238422394,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32470703125,
        "ot_gate_min": 0.0263671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 0.03434963524341583,
        "flow_mse": 0.03434963524341583,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2376708984375,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 0.03885306790471077,
        "flow_mse": 0.03885306790471077,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.258056640625,
        "ot_gate_min": 0.02117919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 0.09729008376598358,
        "flow_mse": 0.09729008376598358,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.362548828125,
        "ot_gate_min": 0.0225677490234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 0.0418848842382431,
        "flow_mse": 0.0418848842382431,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.388671875,
        "ot_gate_min": 0.0225830078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 0.02711518108844757,
        "flow_mse": 0.02711518108844757,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.377685546875,
        "ot_gate_min": 0.0220947265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 0.06759028136730194,
        "flow_mse": 0.06759028136730194,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2430419921875,
        "ot_gate_min": 0.0213470458984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 0.04176453873515129,
        "flow_mse": 0.04176453873515129,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.154541015625,
        "ot_gate_min": 0.02166748046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 0.03798762708902359,
        "flow_mse": 0.03798762708902359,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1387939453125,
        "ot_gate_min": 0.02191162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 0.03845261037349701,
        "flow_mse": 0.03845261037349701,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.257568359375,
        "ot_gate_min": 0.021240234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 0.027387026697397232,
        "flow_mse": 0.027387026697397232,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1409912109375,
        "ot_gate_min": 0.021881103515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 0.06707841157913208,
        "flow_mse": 0.06707841157913208,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.15576171875,
        "ot_gate_min": 0.02239990234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 0.04843991622328758,
        "flow_mse": 0.04843991622328758,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.18798828125,
        "ot_gate_min": 0.0215606689453125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 21,
        "loss": 0.11064091324806213,
        "flow_mse": 0.11064091324806213,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.251220703125,
        "ot_gate_min": 0.0213775634765625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 22,
        "loss": 0.09276309609413147,
        "flow_mse": 0.09276309609413147,
        "t_mean": 0.5307409167289734,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.347900390625,
        "ot_gate_min": 0.02581787109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 23,
        "loss": 0.10600399971008301,
        "flow_mse": 0.10600399971008301,
        "t_mean": 0.511449933052063,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283935546875,
        "ot_gate_min": 0.025390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 24,
        "loss": 0.05860128998756409,
        "flow_mse": 0.05860128998756409,
        "t_mean": 0.5524937510490417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.269775390625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 25,
        "loss": 0.04935474693775177,
        "flow_mse": 0.04935474693775177,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2459716796875,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 26,
        "loss": 0.038127604871988297,
        "flow_mse": 0.038127604871988297,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39306640625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 27,
        "loss": 0.03302235156297684,
        "flow_mse": 0.03302235156297684,
        "t_mean": 0.7675442695617676,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.321044921875,
        "ot_gate_min": 0.0235137939453125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 28,
        "loss": 0.15695452690124512,
        "flow_mse": 0.15695452690124512,
        "t_mean": 0.6012675166130066,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.222412109375,
        "ot_gate_min": 0.0216522216796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 29,
        "loss": 0.028564879670739174,
        "flow_mse": 0.028564879670739174,
        "t_mean": 0.6016367673873901,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.260986328125,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 30,
        "loss": 0.053190842270851135,
        "flow_mse": 0.053190842270851135,
        "t_mean": 0.5350275039672852,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.17578125,
        "ot_gate_min": 0.0223236083984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 31,
        "loss": 0.040011756122112274,
        "flow_mse": 0.040011756122112274,
        "t_mean": 0.3996628522872925,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283203125,
        "ot_gate_min": 0.0210723876953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 32,
        "loss": 0.04932916909456253,
        "flow_mse": 0.04932916909456253,
        "t_mean": 0.4242017865180969,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.268310546875,
        "ot_gate_min": 0.0214080810546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 33,
        "loss": 0.03469334542751312,
        "flow_mse": 0.03469334542751312,
        "t_mean": 0.6198998689651489,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39453125,
        "ot_gate_min": 0.021881103515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 34,
        "loss": 0.03624699264764786,
        "flow_mse": 0.03624699264764786,
        "t_mean": 0.4256276786327362,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283203125,
        "ot_gate_min": 0.0211029052734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 35,
        "loss": 0.04187341779470444,
        "flow_mse": 0.04187341779470444,
        "t_mean": 0.5468468070030212,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1812744140625,
        "ot_gate_min": 0.0212860107421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 36,
        "loss": 0.09272646903991699,
        "flow_mse": 0.09272646903991699,
        "t_mean": 0.4543689489364624,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.18701171875,
        "ot_gate_min": 0.0214080810546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 37,
        "loss": 0.13329653441905975,
        "flow_mse": 0.13329653441905975,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.22314453125,
        "ot_gate_min": 0.0219268798828125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 38,
        "loss": 0.08109062165021896,
        "flow_mse": 0.08109062165021896,
        "t_mean": 0.4061962068080902,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.273681640625,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 39,
        "loss": 0.04787556827068329,
        "flow_mse": 0.04787556827068329,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2509765625,
        "ot_gate_min": 0.0219268798828125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 40,
        "loss": 0.03859013691544533,
        "flow_mse": 0.03859013691544533,
        "t_mean": 0.4235572814941406,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.403076171875,
        "ot_gate_min": 0.0218048095703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 41,
        "loss": 0.08707265555858612,
        "flow_mse": 0.08707265555858612,
        "t_mean": 0.3948693871498108,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3564453125,
        "ot_gate_min": 0.0226593017578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 42,
        "loss": 0.19932441413402557,
        "flow_mse": 0.19932441413402557,
        "t_mean": 0.5838073492050171,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26806640625,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 43,
        "loss": 0.053420405834913254,
        "flow_mse": 0.053420405834913254,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26904296875,
        "ot_gate_min": 0.02099609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 44,
        "loss": 0.050102781504392624,
        "flow_mse": 0.050102781504392624,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26611328125,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 45,
        "loss": 0.02903689816594124,
        "flow_mse": 0.02903689816594124,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.372802734375,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 46,
        "loss": 0.06426116824150085,
        "flow_mse": 0.06426116824150085,
        "t_mean": 0.3058706820011139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.192626953125,
        "ot_gate_min": 0.021514892578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 47,
        "loss": 0.049310266971588135,
        "flow_mse": 0.049310266971588135,
        "t_mean": 0.31310853362083435,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1556396484375,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 48,
        "loss": 0.06352117657661438,
        "flow_mse": 0.06352117657661438,
        "t_mean": 0.5626304149627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.388916015625,
        "ot_gate_min": 0.0229949951171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 49,
        "loss": 0.05685967206954956,
        "flow_mse": 0.05685967206954956,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283447265625,
        "ot_gate_min": 0.0210113525390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 50,
        "loss": 0.04721525311470032,
        "flow_mse": 0.04721525311470032,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.27783203125,
        "ot_gate_min": 0.0210723876953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 51,
        "loss": 0.05415944382548332,
        "flow_mse": 0.05415944382548332,
        "t_mean": 0.6815006732940674,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.38134765625,
        "ot_gate_min": 0.0229644775390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 52,
        "loss": 0.03573838993906975,
        "flow_mse": 0.03573838993906975,
        "t_mean": 0.509903609752655,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39306640625,
        "ot_gate_min": 0.02294921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 53,
        "loss": 0.022481029853224754,
        "flow_mse": 0.022481029853224754,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.394287109375,
        "ot_gate_min": 0.02203369140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 54,
        "loss": 0.15063808858394623,
        "flow_mse": 0.15063808858394623,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2252197265625,
        "ot_gate_min": 0.0219879150390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 55,
        "loss": 0.12002533674240112,
        "flow_mse": 0.12002533674240112,
        "t_mean": 0.7171943187713623,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.251708984375,
        "ot_gate_min": 0.0214996337890625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 56,
        "loss": 0.08033160120248795,
        "flow_mse": 0.08033160120248795,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.257080078125,
        "ot_gate_min": 0.0213775634765625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 57,
        "loss": 0.038511358201503754,
        "flow_mse": 0.038511358201503754,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.394287109375,
        "ot_gate_min": 0.023681640625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 58,
        "loss": 0.03624273091554642,
        "flow_mse": 0.03624273091554642,
        "t_mean": 0.5147815942764282,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.288818359375,
        "ot_gate_min": 0.02093505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 59,
        "loss": 0.09758208692073822,
        "flow_mse": 0.09758208692073822,
        "t_mean": 0.49480095505714417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.292236328125,
        "ot_gate_min": 0.025177001953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 60,
        "loss": 0.04789481684565544,
        "flow_mse": 0.04789481684565544,
        "t_mean": 0.4051949977874756,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.0,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.268310546875,
        "ot_gate_min": 0.021575927734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Official full Stage1 RGB edge",
    "group": "stage1",
    "phase": "official-full",
    "note": "Official-initialized full matrix with RGB Sobel edge-regularized OT cost.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage1_rgb_edge",
    "status": "ok",
    "steps": 60,
    "loss": 0.0478987880051136,
    "flow_mse": 0.0478987880051136,
    "art_l2": null,
    "ot_gate_mean": 0.2685546875,
    "ot_edge_delta": 0.0011039308737963438,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage1_rgb_edge",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.02921142242848873,
        "flow_mse": 0.02921142242848873,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001398545573465526,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29248046875,
        "ot_gate_min": 0.0208740234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 0.06307670474052429,
        "flow_mse": 0.06307670474052429,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010306560434401035,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1976318359375,
        "ot_gate_min": 0.022918701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 0.06076542288064957,
        "flow_mse": 0.06076542288064957,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009209124837070704,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3642578125,
        "ot_gate_min": 0.0224456787109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 0.043031394481658936,
        "flow_mse": 0.043031394481658936,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001137981773354113,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26806640625,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 0.049725405871868134,
        "flow_mse": 0.049725405871868134,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010986243141815066,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2308349609375,
        "ot_gate_min": 0.0218353271484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 0.05110844969749451,
        "flow_mse": 0.05110844969749451,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012084253830835223,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2310791015625,
        "ot_gate_min": 0.0216217041015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 0.05507455766201019,
        "flow_mse": 0.05507455766201019,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010872404091060162,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.33056640625,
        "ot_gate_min": 0.0217132568359375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 0.10008716583251953,
        "flow_mse": 0.10008716583251953,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011022890685126185,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32470703125,
        "ot_gate_min": 0.0263671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 0.03435647860169411,
        "flow_mse": 0.03435647860169411,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001422184519469738,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2379150390625,
        "ot_gate_min": 0.0217742919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 0.03875892609357834,
        "flow_mse": 0.03875892609357834,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001209727139212191,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.258544921875,
        "ot_gate_min": 0.02117919921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 0.09728798270225525,
        "flow_mse": 0.09728798270225525,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011889992747455835,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.362548828125,
        "ot_gate_min": 0.0225677490234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 0.041876401752233505,
        "flow_mse": 0.041876401752233505,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001179387210868299,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.388671875,
        "ot_gate_min": 0.0225830078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 0.02712138183414936,
        "flow_mse": 0.02712138183414936,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010838002199307084,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3779296875,
        "ot_gate_min": 0.0221099853515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 0.06757485866546631,
        "flow_mse": 0.06757485866546631,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.000977051560766995,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2432861328125,
        "ot_gate_min": 0.0213623046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 0.041762713342905045,
        "flow_mse": 0.041762713342905045,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012366290902718902,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.15478515625,
        "ot_gate_min": 0.02166748046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 0.03798414394259453,
        "flow_mse": 0.03798414394259453,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010139193618670106,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1390380859375,
        "ot_gate_min": 0.02191162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 0.03845524415373802,
        "flow_mse": 0.03845524415373802,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.00099812017288059,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.257568359375,
        "ot_gate_min": 0.021240234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 0.02738991379737854,
        "flow_mse": 0.02738991379737854,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011173380771651864,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.14111328125,
        "ot_gate_min": 0.021881103515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 0.06707742810249329,
        "flow_mse": 0.06707742810249329,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010231205960735679,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1558837890625,
        "ot_gate_min": 0.0224151611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 0.04843730852007866,
        "flow_mse": 0.04843730852007866,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013149974402040243,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.18798828125,
        "ot_gate_min": 0.0215606689453125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 21,
        "loss": 0.11063931882381439,
        "flow_mse": 0.11063931882381439,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011202176101505756,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25146484375,
        "ot_gate_min": 0.0213775634765625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 22,
        "loss": 0.09276279807090759,
        "flow_mse": 0.09276279807090759,
        "t_mean": 0.5307409167289734,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011577386176213622,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.347900390625,
        "ot_gate_min": 0.02581787109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 23,
        "loss": 0.10599254071712494,
        "flow_mse": 0.10599254071712494,
        "t_mean": 0.511449933052063,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0008798614726401865,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2841796875,
        "ot_gate_min": 0.025421142578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 24,
        "loss": 0.058603812009096146,
        "flow_mse": 0.058603812009096146,
        "t_mean": 0.5524937510490417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010997080244123936,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.269775390625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 25,
        "loss": 0.049354564398527145,
        "flow_mse": 0.049354564398527145,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009859190322458744,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.24609375,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 26,
        "loss": 0.03812946006655693,
        "flow_mse": 0.03812946006655693,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011186831397935748,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.394775390625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 27,
        "loss": 0.03305887430906296,
        "flow_mse": 0.03305887430906296,
        "t_mean": 0.7675442695617676,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009447630727663636,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3212890625,
        "ot_gate_min": 0.0235443115234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 28,
        "loss": 0.15695305168628693,
        "flow_mse": 0.15695305168628693,
        "t_mean": 0.6012675166130066,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012292672181501985,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.222412109375,
        "ot_gate_min": 0.0216522216796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 29,
        "loss": 0.02857532538473606,
        "flow_mse": 0.02857532538473606,
        "t_mean": 0.6016367673873901,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011290234979242086,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26123046875,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 30,
        "loss": 0.053201474249362946,
        "flow_mse": 0.053201474249362946,
        "t_mean": 0.5350275039672852,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012161594349890947,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.176025390625,
        "ot_gate_min": 0.0223236083984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 31,
        "loss": 0.04001445695757866,
        "flow_mse": 0.04001445695757866,
        "t_mean": 0.3996628522872925,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010223782155662775,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283203125,
        "ot_gate_min": 0.0210723876953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 32,
        "loss": 0.049333032220602036,
        "flow_mse": 0.049333032220602036,
        "t_mean": 0.4242017865180969,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014547370374202728,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2685546875,
        "ot_gate_min": 0.0214080810546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 33,
        "loss": 0.03469755873084068,
        "flow_mse": 0.03469755873084068,
        "t_mean": 0.6198998689651489,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011624412145465612,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39501953125,
        "ot_gate_min": 0.021881103515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 34,
        "loss": 0.036246418952941895,
        "flow_mse": 0.036246418952941895,
        "t_mean": 0.4256276786327362,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010150743182748556,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283203125,
        "ot_gate_min": 0.0211029052734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 35,
        "loss": 0.04187668114900589,
        "flow_mse": 0.04187668114900589,
        "t_mean": 0.5468468070030212,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011868680594488978,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1815185546875,
        "ot_gate_min": 0.0212860107421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 36,
        "loss": 0.09272445738315582,
        "flow_mse": 0.09272445738315582,
        "t_mean": 0.4543689489364624,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014237647410482168,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.18701171875,
        "ot_gate_min": 0.0214080810546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 37,
        "loss": 0.1330461949110031,
        "flow_mse": 0.1330461949110031,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010284001473337412,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.223876953125,
        "ot_gate_min": 0.021942138671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 38,
        "loss": 0.08108960837125778,
        "flow_mse": 0.08108960837125778,
        "t_mean": 0.4061962068080902,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014305978547781706,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.274169921875,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 39,
        "loss": 0.04787101969122887,
        "flow_mse": 0.04787101969122887,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009340504184365273,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2509765625,
        "ot_gate_min": 0.0219268798828125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 40,
        "loss": 0.03858843445777893,
        "flow_mse": 0.03858843445777893,
        "t_mean": 0.4235572814941406,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012591193662956357,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.403076171875,
        "ot_gate_min": 0.0218048095703125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 41,
        "loss": 0.08707933127880096,
        "flow_mse": 0.08707933127880096,
        "t_mean": 0.3948693871498108,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0008600704022683203,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3564453125,
        "ot_gate_min": 0.022674560546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 42,
        "loss": 0.1993253231048584,
        "flow_mse": 0.1993253231048584,
        "t_mean": 0.5838073492050171,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013301416765898466,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.268798828125,
        "ot_gate_min": 0.021484375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 43,
        "loss": 0.053427062928676605,
        "flow_mse": 0.053427062928676605,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010882187634706497,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.269287109375,
        "ot_gate_min": 0.02099609375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 44,
        "loss": 0.0500980019569397,
        "flow_mse": 0.0500980019569397,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001183810643851757,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.266357421875,
        "ot_gate_min": 0.0211944580078125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 45,
        "loss": 0.02903621271252632,
        "flow_mse": 0.02903621271252632,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011854786425828934,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.372802734375,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 46,
        "loss": 0.06425946950912476,
        "flow_mse": 0.06425946950912476,
        "t_mean": 0.3058706820011139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014449775917455554,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.19287109375,
        "ot_gate_min": 0.021514892578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 47,
        "loss": 0.04931196942925453,
        "flow_mse": 0.04931196942925453,
        "t_mean": 0.31310853362083435,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012737808283418417,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1556396484375,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 48,
        "loss": 0.06352312862873077,
        "flow_mse": 0.06352312862873077,
        "t_mean": 0.5626304149627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0008127482142299414,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3896484375,
        "ot_gate_min": 0.0229949951171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 49,
        "loss": 0.056857477873563766,
        "flow_mse": 0.056857477873563766,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012269875733181834,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.283447265625,
        "ot_gate_min": 0.0210113525390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 50,
        "loss": 0.047210365533828735,
        "flow_mse": 0.047210365533828735,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011668781517073512,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.27783203125,
        "ot_gate_min": 0.0210723876953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 51,
        "loss": 0.05416306108236313,
        "flow_mse": 0.05416306108236313,
        "t_mean": 0.6815006732940674,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010999992955476046,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3818359375,
        "ot_gate_min": 0.0229644775390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 52,
        "loss": 0.035737358033657074,
        "flow_mse": 0.035737358033657074,
        "t_mean": 0.509903609752655,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001130940392613411,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39306640625,
        "ot_gate_min": 0.02294921875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 53,
        "loss": 0.022479940205812454,
        "flow_mse": 0.022479940205812454,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.00111474993173033,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.395263671875,
        "ot_gate_min": 0.02203369140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 54,
        "loss": 0.15063321590423584,
        "flow_mse": 0.15063321590423584,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011199887376278639,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2254638671875,
        "ot_gate_min": 0.0219879150390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 55,
        "loss": 0.11999987065792084,
        "flow_mse": 0.11999987065792084,
        "t_mean": 0.7171943187713623,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010693680960685015,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.251708984375,
        "ot_gate_min": 0.0214996337890625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 56,
        "loss": 0.0803375244140625,
        "flow_mse": 0.0803375244140625,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011400202056393027,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25732421875,
        "ot_gate_min": 0.0213775634765625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 57,
        "loss": 0.03851146996021271,
        "flow_mse": 0.03851146996021271,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010651889024302363,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.395263671875,
        "ot_gate_min": 0.023681640625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 58,
        "loss": 0.03624003380537033,
        "flow_mse": 0.03624003380537033,
        "t_mean": 0.5147815942764282,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009497925057075918,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.289306640625,
        "ot_gate_min": 0.02093505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 59,
        "loss": 0.09757418930530548,
        "flow_mse": 0.09757418930530548,
        "t_mean": 0.49480095505714417,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010648122988641262,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.292236328125,
        "ot_gate_min": 0.025177001953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 60,
        "loss": 0.0478987880051136,
        "flow_mse": 0.0478987880051136,
        "t_mean": 0.4051949977874756,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011039308737963438,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2685546875,
        "ot_gate_min": 0.021575927734375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Official full Stage1 virtual patch",
    "group": "stage1",
    "phase": "official-full",
    "note": "Official-initialized full matrix with a virtual patch channel.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage1_virtual_patch",
    "status": "ok",
    "steps": 60,
    "loss": 0.05484720319509506,
    "flow_mse": 0.05484720319509506,
    "art_l2": null,
    "ot_gate_mean": 0.286376953125,
    "ot_edge_delta": 0.0009899159194901586,
    "ot_virtual_mass": 0.04998779296875,
    "ot_modules_active": 1.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "rgb",
      "virtual_tokens": 1,
      "virtual_mass": 0.05,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage1_virtual_patch",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 60,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "rgb",
      "ot_virtual_tokens": 1,
      "ot_virtual_mass": 0.05,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.026793912053108215,
        "flow_mse": 0.026793912053108215,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009518907754682004,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.4013671875,
        "ot_gate_min": 0.02166748046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 2,
        "loss": 0.05393337458372116,
        "flow_mse": 0.05393337458372116,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010577188804745674,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.216064453125,
        "ot_gate_min": 0.0223236083984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 3,
        "loss": 0.05763113498687744,
        "flow_mse": 0.05763113498687744,
        "t_mean": 0.6866899728775024,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009937018621712923,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.156494140625,
        "ot_gate_min": 0.02197265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 4,
        "loss": 0.03923727944493294,
        "flow_mse": 0.03923727944493294,
        "t_mean": 0.5447205901145935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001060508075170219,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1578369140625,
        "ot_gate_min": 0.0216522216796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 5,
        "loss": 0.037792809307575226,
        "flow_mse": 0.037792809307575226,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011901373509317636,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2607421875,
        "ot_gate_min": 0.021331787109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 6,
        "loss": 0.09327708184719086,
        "flow_mse": 0.09327708184719086,
        "t_mean": 0.681830883026123,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010309215867891908,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.293212890625,
        "ot_gate_min": 0.025390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 7,
        "loss": 0.08595950901508331,
        "flow_mse": 0.08595950901508331,
        "t_mean": 0.7412480115890503,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001144148176535964,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.345458984375,
        "ot_gate_min": 0.025390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 8,
        "loss": 0.10656894743442535,
        "flow_mse": 0.10656894743442535,
        "t_mean": 0.39051905274391174,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012174316216260195,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.14453125,
        "ot_gate_min": 0.0218505859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 9,
        "loss": 0.029916221275925636,
        "flow_mse": 0.029916221275925636,
        "t_mean": 0.5102756023406982,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0016218394739553332,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1683349609375,
        "ot_gate_min": 0.0216522216796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 10,
        "loss": 0.21641503274440765,
        "flow_mse": 0.21641503274440765,
        "t_mean": 0.5520592927932739,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013210710603743792,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.266357421875,
        "ot_gate_min": 0.0216217041015625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 11,
        "loss": 0.03778097778558731,
        "flow_mse": 0.03778097778558731,
        "t_mean": 0.593870222568512,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0014373642625287175,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.277587890625,
        "ot_gate_min": 0.021453857421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 12,
        "loss": 0.03370743244886398,
        "flow_mse": 0.03370743244886398,
        "t_mean": 0.42200398445129395,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009971901308745146,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.379638671875,
        "ot_gate_min": 0.0223388671875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 13,
        "loss": 0.03794397413730621,
        "flow_mse": 0.03794397413730621,
        "t_mean": 0.39839279651641846,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010991119779646397,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26318359375,
        "ot_gate_min": 0.021453857421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 14,
        "loss": 0.08168123662471771,
        "flow_mse": 0.08168123662471771,
        "t_mean": 0.581339418888092,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010169727029278874,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.381591796875,
        "ot_gate_min": 0.0244293212890625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 15,
        "loss": 0.041838787496089935,
        "flow_mse": 0.041838787496089935,
        "t_mean": 0.3876829147338867,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009499876177869737,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.390869140625,
        "ot_gate_min": 0.0227203369140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 16,
        "loss": 0.09880691021680832,
        "flow_mse": 0.09880691021680832,
        "t_mean": 0.5303117036819458,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001106880372390151,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.323974609375,
        "ot_gate_min": 0.0261077880859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 17,
        "loss": 0.038283172994852066,
        "flow_mse": 0.038283172994852066,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010865225922316313,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.265380859375,
        "ot_gate_min": 0.0212249755859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 18,
        "loss": 0.03842104226350784,
        "flow_mse": 0.03842104226350784,
        "t_mean": 0.4426919221878052,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011755574960261583,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1732177734375,
        "ot_gate_min": 0.022125244140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 19,
        "loss": 0.07571597397327423,
        "flow_mse": 0.07571597397327423,
        "t_mean": 0.4035862386226654,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0008248382364399731,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.39306640625,
        "ot_gate_min": 0.022979736328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 20,
        "loss": 0.032571516931056976,
        "flow_mse": 0.032571516931056976,
        "t_mean": 0.52027428150177,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010929068084806204,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26318359375,
        "ot_gate_min": 0.02130126953125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 21,
        "loss": 0.07668634504079819,
        "flow_mse": 0.07668634504079819,
        "t_mean": 0.31838786602020264,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.00122437858954072,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26416015625,
        "ot_gate_min": 0.0212249755859375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 22,
        "loss": 0.045333489775657654,
        "flow_mse": 0.045333489775657654,
        "t_mean": 0.5029113292694092,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001319806557148695,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1605224609375,
        "ot_gate_min": 0.0215911865234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 23,
        "loss": 0.04698977991938591,
        "flow_mse": 0.04698977991938591,
        "t_mean": 0.45976296067237854,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001199997030198574,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2587890625,
        "ot_gate_min": 0.021759033203125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 24,
        "loss": 0.09754927456378937,
        "flow_mse": 0.09754927456378937,
        "t_mean": 0.7851753234863281,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011665665078908205,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32275390625,
        "ot_gate_min": 0.02587890625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 25,
        "loss": 0.05402711033821106,
        "flow_mse": 0.05402711033821106,
        "t_mean": 0.4828733801841736,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001133846351876855,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.264404296875,
        "ot_gate_min": 0.0214080810546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 26,
        "loss": 0.040744006633758545,
        "flow_mse": 0.040744006633758545,
        "t_mean": 0.47976693511009216,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012145057553425431,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.231689453125,
        "ot_gate_min": 0.021881103515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 27,
        "loss": 0.0906679630279541,
        "flow_mse": 0.0906679630279541,
        "t_mean": 0.5499228239059448,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009777913801372051,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.239990234375,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 28,
        "loss": 0.08841349929571152,
        "flow_mse": 0.08841349929571152,
        "t_mean": 0.5113206505775452,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009849605849012733,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3359375,
        "ot_gate_min": 0.0238494873046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 29,
        "loss": 0.04032602533698082,
        "flow_mse": 0.04032602533698082,
        "t_mean": 0.5222197771072388,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011214644182473421,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.38330078125,
        "ot_gate_min": 0.0229644775390625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 30,
        "loss": 0.07197731733322144,
        "flow_mse": 0.07197731733322144,
        "t_mean": 0.5769186019897461,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.00117973901797086,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25732421875,
        "ot_gate_min": 0.0212860107421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 31,
        "loss": 0.09193521738052368,
        "flow_mse": 0.09193521738052368,
        "t_mean": 0.314164400100708,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001140001229941845,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.389404296875,
        "ot_gate_min": 0.02313232421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 32,
        "loss": 0.041706643998622894,
        "flow_mse": 0.041706643998622894,
        "t_mean": 0.5159284472465515,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.00138756912201643,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26953125,
        "ot_gate_min": 0.021209716796875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 33,
        "loss": 0.03009115718305111,
        "flow_mse": 0.03009115718305111,
        "t_mean": 0.5761235952377319,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011088825995102525,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.261474609375,
        "ot_gate_min": 0.02142333984375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 34,
        "loss": 0.09237630665302277,
        "flow_mse": 0.09237630665302277,
        "t_mean": 0.3432917296886444,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011321635683998466,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.398193359375,
        "ot_gate_min": 0.02557373046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 35,
        "loss": 0.0923071801662445,
        "flow_mse": 0.0923071801662445,
        "t_mean": 0.5136957168579102,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010895878076553345,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.384033203125,
        "ot_gate_min": 0.024078369140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 36,
        "loss": 0.07200980931520462,
        "flow_mse": 0.07200980931520462,
        "t_mean": 0.449867308139801,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011279706377536058,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.267333984375,
        "ot_gate_min": 0.021728515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 37,
        "loss": 0.056326724588871,
        "flow_mse": 0.056326724588871,
        "t_mean": 0.5334687829017639,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011718154419213533,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2587890625,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 38,
        "loss": 0.055778078734874725,
        "flow_mse": 0.055778078734874725,
        "t_mean": 0.49369919300079346,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001333719352260232,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.187744140625,
        "ot_gate_min": 0.0215606689453125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 39,
        "loss": 0.03684721142053604,
        "flow_mse": 0.03684721142053604,
        "t_mean": 0.6304954886436462,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010232412023469806,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2841796875,
        "ot_gate_min": 0.021240234375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 40,
        "loss": 0.027343830093741417,
        "flow_mse": 0.027343830093741417,
        "t_mean": 0.5427849888801575,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011795543832704425,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.371337890625,
        "ot_gate_min": 0.0220947265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 41,
        "loss": 0.055506013333797455,
        "flow_mse": 0.055506013333797455,
        "t_mean": 0.6182063817977905,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011538758408278227,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2281494140625,
        "ot_gate_min": 0.021697998046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 42,
        "loss": 0.04832226037979126,
        "flow_mse": 0.04832226037979126,
        "t_mean": 0.5307261943817139,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013578878715634346,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1309814453125,
        "ot_gate_min": 0.0217437744140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 43,
        "loss": 0.04029387980699539,
        "flow_mse": 0.04029387980699539,
        "t_mean": 0.674645185470581,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009180419147014618,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.274658203125,
        "ot_gate_min": 0.021026611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 44,
        "loss": 0.051952000707387924,
        "flow_mse": 0.051952000707387924,
        "t_mean": 0.4860682487487793,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0012182778446003795,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.251220703125,
        "ot_gate_min": 0.021392822265625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 45,
        "loss": 0.04857983440160751,
        "flow_mse": 0.04857983440160751,
        "t_mean": 0.3828374445438385,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.000996182905510068,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.364990234375,
        "ot_gate_min": 0.0226593017578125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 46,
        "loss": 0.050921183079481125,
        "flow_mse": 0.050921183079481125,
        "t_mean": 0.3461093604564667,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001118855201639235,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1578369140625,
        "ot_gate_min": 0.021728515625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 47,
        "loss": 0.08714324235916138,
        "flow_mse": 0.08714324235916138,
        "t_mean": 0.370074987411499,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011027801083400846,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.255126953125,
        "ot_gate_min": 0.0214691162109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 48,
        "loss": 0.06117895245552063,
        "flow_mse": 0.06117895245552063,
        "t_mean": 0.6085288524627686,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013406730722635984,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1661376953125,
        "ot_gate_min": 0.0218963623046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 49,
        "loss": 0.1375504583120346,
        "flow_mse": 0.1375504583120346,
        "t_mean": 0.7366573214530945,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010867038508877158,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2215576171875,
        "ot_gate_min": 0.0224761962890625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 50,
        "loss": 0.03477924317121506,
        "flow_mse": 0.03477924317121506,
        "t_mean": 0.5806166529655457,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001009408850222826,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2049560546875,
        "ot_gate_min": 0.02276611328125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 51,
        "loss": 0.0812399834394455,
        "flow_mse": 0.0812399834394455,
        "t_mean": 0.5388283729553223,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0013958263443782926,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1868896484375,
        "ot_gate_min": 0.0214080810546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 52,
        "loss": 0.10480914264917374,
        "flow_mse": 0.10480914264917374,
        "t_mean": 0.5041043758392334,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010339601431041956,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29345703125,
        "ot_gate_min": 0.0253143310546875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 53,
        "loss": 0.035486478358507156,
        "flow_mse": 0.035486478358507156,
        "t_mean": 0.5933268070220947,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0008270475082099438,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2015380859375,
        "ot_gate_min": 0.0228729248046875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 54,
        "loss": 0.05259774252772331,
        "flow_mse": 0.05259774252772331,
        "t_mean": 0.5004596710205078,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011861682869493961,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.265380859375,
        "ot_gate_min": 0.0212860107421875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 55,
        "loss": 0.08578689396381378,
        "flow_mse": 0.08578689396381378,
        "t_mean": 0.6222312450408936,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010737701086327434,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.349853515625,
        "ot_gate_min": 0.02264404296875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 56,
        "loss": 0.06720918416976929,
        "flow_mse": 0.06720918416976929,
        "t_mean": 0.4547177255153656,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0011338602053001523,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.40625,
        "ot_gate_min": 0.021331787109375,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 57,
        "loss": 0.10348159074783325,
        "flow_mse": 0.10348159074783325,
        "t_mean": 0.43234071135520935,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.001497256918810308,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.393798828125,
        "ot_gate_min": 0.0244140625,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 58,
        "loss": 0.05218612402677536,
        "flow_mse": 0.05218612402677536,
        "t_mean": 0.4430321455001831,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010405932553112507,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2333984375,
        "ot_gate_min": 0.0219268798828125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.050018310546875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 59,
        "loss": 0.07473547756671906,
        "flow_mse": 0.07473547756671906,
        "t_mean": 0.45887431502342224,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0010784438345581293,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.251953125,
        "ot_gate_min": 0.0216064453125,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      },
      {
        "step": 60,
        "loss": 0.05484720319509506,
        "flow_mse": 0.05484720319509506,
        "t_mean": 0.46173495054244995,
        "ot_modules_active": 1.0,
        "ot_edge_delta": 0.0009899159194901586,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.4,
        "ot_edge_source_rgb": 1.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.286376953125,
        "ot_gate_min": 0.0210418701171875,
        "ot_mask_prior_beta": 0.2,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.04998779296875,
        "ot_virtual_tokens": 1.0
      }
    ]
  },
  {
    "label": "Official full Stage1 first-third edge",
    "group": "stage1",
    "phase": "official-full",
    "note": "Official-initialized full matrix with OT injected into the first third of Stage1 blocks.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage1_firstthird_edge",
    "status": "ok",
    "steps": 20,
    "loss": 0.04857778549194336,
    "flow_mse": 0.04857778549194336,
    "art_l2": null,
    "ot_gate_mean": 0.184539794921875,
    "ot_edge_delta": 0.0018430916970828548,
    "ot_virtual_mass": 0.0,
    "ot_modules_active": 8.0,
    "stage2_mask_part_count": null,
    "stage2_mask_empty_parts": null,
    "stage2_mask_centroid_spread": null,
    "stage2_mask_spatial_var_mean": null,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": true,
      "block_selector": "first-third",
      "sinkhorn_iters": 10,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.4,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.2
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage1",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage1_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage1_firstthird_edge",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/part_sparse_structure_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 20,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": false,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": true,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 10,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.4,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.2
    },
    "logs": [
      {
        "step": 1,
        "loss": 0.029235010966658592,
        "flow_mse": 0.029235010966658592,
        "t_mean": 0.4452204704284668,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0015091553650563583,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.29052734375,
        "ot_gate_min": 0.020868301391601562,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 2,
        "loss": 0.06334857642650604,
        "flow_mse": 0.06334857642650604,
        "t_mean": 0.25286394357681274,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0015698453353252262,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1953277587890625,
        "ot_gate_min": 0.022754669189453125,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 3,
        "loss": 0.06327332556247711,
        "flow_mse": 0.06327332556247711,
        "t_mean": 0.6159385442733765,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001538539188914001,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.359466552734375,
        "ot_gate_min": 0.0220794677734375,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 4,
        "loss": 0.04134778678417206,
        "flow_mse": 0.04134778678417206,
        "t_mean": 0.6475216150283813,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0016027584060793743,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.26055908203125,
        "ot_gate_min": 0.021097183227539062,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 5,
        "loss": 0.05018697679042816,
        "flow_mse": 0.05018697679042816,
        "t_mean": 0.6798597574234009,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0015820720582269132,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.224761962890625,
        "ot_gate_min": 0.021669387817382812,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 6,
        "loss": 0.05024827644228935,
        "flow_mse": 0.05024827644228935,
        "t_mean": 0.5003219842910767,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0017732312844600528,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.224456787109375,
        "ot_gate_min": 0.02166748046875,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 7,
        "loss": 0.05355503782629967,
        "flow_mse": 0.05355503782629967,
        "t_mean": 0.6674641966819763,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0016843462799442932,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.316162109375,
        "ot_gate_min": 0.0213470458984375,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 8,
        "loss": 0.0988912582397461,
        "flow_mse": 0.0988912582397461,
        "t_mean": 0.6387549638748169,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.001548043088405393,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.32244873046875,
        "ot_gate_min": 0.02593231201171875,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 9,
        "loss": 0.03443935886025429,
        "flow_mse": 0.03443935886025429,
        "t_mean": 0.5298079252243042,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0019439083553152159,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.2297515869140625,
        "ot_gate_min": 0.02169036865234375,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 10,
        "loss": 0.03599346801638603,
        "flow_mse": 0.03599346801638603,
        "t_mean": 0.5381855368614197,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0015831510681891814,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.252838134765625,
        "ot_gate_min": 0.021177291870117188,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 11,
        "loss": 0.0994499921798706,
        "flow_mse": 0.0994499921798706,
        "t_mean": 0.5974305868148804,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0017243401816813275,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.35821533203125,
        "ot_gate_min": 0.022369384765625,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 12,
        "loss": 0.041969697922468185,
        "flow_mse": 0.041969697922468185,
        "t_mean": 0.4660314917564392,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0018068547215079889,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.378570556640625,
        "ot_gate_min": 0.022394180297851562,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 13,
        "loss": 0.027066143229603767,
        "flow_mse": 0.027066143229603767,
        "t_mean": 0.42932528257369995,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0016399096639361233,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.3731689453125,
        "ot_gate_min": 0.021947860717773438,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 14,
        "loss": 0.06713995337486267,
        "flow_mse": 0.06713995337486267,
        "t_mean": 0.44856566190719604,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0014165033790050074,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.241485595703125,
        "ot_gate_min": 0.02136993408203125,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 15,
        "loss": 0.041769251227378845,
        "flow_mse": 0.041769251227378845,
        "t_mean": 0.34305432438850403,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.00173603804432787,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.153076171875,
        "ot_gate_min": 0.021696090698242188,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 16,
        "loss": 0.03708484768867493,
        "flow_mse": 0.03708484768867493,
        "t_mean": 0.4998842179775238,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0016036720044212416,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.137664794921875,
        "ot_gate_min": 0.021839141845703125,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 17,
        "loss": 0.037994664162397385,
        "flow_mse": 0.037994664162397385,
        "t_mean": 0.4249878525733948,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0016522705554962158,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.25341796875,
        "ot_gate_min": 0.021116256713867188,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 18,
        "loss": 0.027192318812012672,
        "flow_mse": 0.027192318812012672,
        "t_mean": 0.42435136437416077,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0015692258020862937,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.1392364501953125,
        "ot_gate_min": 0.021795272827148438,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 19,
        "loss": 0.06675601750612259,
        "flow_mse": 0.06675601750612259,
        "t_mean": 0.49784499406814575,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0014031500031705946,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.153167724609375,
        "ot_gate_min": 0.022218704223632812,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      },
      {
        "step": 20,
        "loss": 0.04857778549194336,
        "flow_mse": 0.04857778549194336,
        "t_mean": 0.4546642005443573,
        "ot_modules_active": 8.0,
        "ot_edge_delta": 0.0018430916970828548,
        "ot_edge_gamma": 8.0,
        "ot_edge_lambda": 0.39999999999999997,
        "ot_edge_source_rgb": 0.0,
        "ot_gate_max": 1.0,
        "ot_gate_mean": 0.184539794921875,
        "ot_gate_min": 0.02144622802734375,
        "ot_mask_prior_beta": 0.19999999999999998,
        "ot_plan_mass": 1.0,
        "ot_virtual_mass": 0.0,
        "ot_virtual_tokens": 0.0
      }
    ]
  },
  {
    "label": "Official full Stage2 spatial diag",
    "group": "stage2",
    "phase": "official-full",
    "note": "Official-initialized full matrix Stage2 spatial diagnostics.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage2_spatial_diag",
    "status": "ok",
    "steps": 40,
    "loss": 1.5618197917938232,
    "flow_mse": 1.4997601509094238,
    "art_l2": 0.06205962225794792,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 10.0,
    "stage2_mask_empty_parts": 3.0,
    "stage2_mask_centroid_spread": 0.12488780915737152,
    "stage2_mask_spatial_var_mean": 0.019491467624902725,
    "stage2_spatial_adapter_l2": null,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": false,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage2_spatial_diag",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 40,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": false,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 2.385606527328491,
        "flow_mse": 2.274975538253784,
        "art_l2": 0.11063092201948166,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 5.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.097297303378582,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1345607340335846,
        "stage2_mask_spatial_var_mean": 0.03559853881597519
      },
      {
        "step": 2,
        "loss": 1.0194467306137085,
        "flow_mse": 0.9345120191574097,
        "art_l2": 0.08493470400571823,
        "t_mean": 0.3323965072631836,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.07617656141519547,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11268322169780731,
        "stage2_mask_spatial_var_mean": 0.024656636640429497
      },
      {
        "step": 3,
        "loss": 4.096953868865967,
        "flow_mse": 4.015328884124756,
        "art_l2": 0.08162512630224228,
        "t_mean": 0.8121371269226074,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.1260043829679489,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13602715730667114,
        "stage2_mask_spatial_var_mean": 0.055630188435316086
      },
      {
        "step": 4,
        "loss": 2.377106189727783,
        "flow_mse": 2.305943250656128,
        "art_l2": 0.07116299867630005,
        "t_mean": 0.686140239238739,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09447285532951355,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12284307181835175,
        "stage2_mask_spatial_var_mean": 0.03497007489204407
      },
      {
        "step": 5,
        "loss": 3.494433641433716,
        "flow_mse": 3.2939414978027344,
        "art_l2": 0.20049221813678741,
        "t_mean": 0.728419840335846,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.13586559891700745,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12117581814527512,
        "stage2_mask_spatial_var_mean": 0.06845814734697342
      },
      {
        "step": 6,
        "loss": 2.5841925144195557,
        "flow_mse": 2.546032190322876,
        "art_l2": 0.03816024959087372,
        "t_mean": 0.6403265595436096,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 9.0,
        "stage2_mask_mass_mean": 0.03964406996965408,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.03179550915956497,
        "stage2_mask_spatial_var_mean": 0.1153540164232254
      },
      {
        "step": 7,
        "loss": 3.98237943649292,
        "flow_mse": 3.855491876602173,
        "art_l2": 0.12688744068145752,
        "t_mean": 0.5700072646141052,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10189920663833618,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17310911417007446,
        "stage2_mask_spatial_var_mean": 0.06178075820207596
      },
      {
        "step": 8,
        "loss": 2.688394546508789,
        "flow_mse": 2.5778989791870117,
        "art_l2": 0.11049561202526093,
        "t_mean": 0.655311644077301,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08826395124197006,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11497963964939117,
        "stage2_mask_spatial_var_mean": 0.030766749754548073
      },
      {
        "step": 9,
        "loss": 3.5994925498962402,
        "flow_mse": 3.4171547889709473,
        "art_l2": 0.18233773112297058,
        "t_mean": 0.6019466519355774,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.1391526758670807,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12259700894355774,
        "stage2_mask_spatial_var_mean": 0.06607501208782196
      },
      {
        "step": 10,
        "loss": 2.757052183151245,
        "flow_mse": 2.583155632019043,
        "art_l2": 0.17389650642871857,
        "t_mean": 0.7785599827766418,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08728999644517899,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.05403045192360878,
        "stage2_mask_spatial_var_mean": 0.056121475994586945
      },
      {
        "step": 11,
        "loss": 4.716871738433838,
        "flow_mse": 4.685192584991455,
        "art_l2": 0.03167920932173729,
        "t_mean": 0.6584867835044861,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11102995276451111,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.22974403202533722,
        "stage2_mask_spatial_var_mean": 0.057314611971378326
      },
      {
        "step": 12,
        "loss": 2.251176357269287,
        "flow_mse": 2.131728172302246,
        "art_l2": 0.11944824457168579,
        "t_mean": 0.2739311456680298,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13604821264743805,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1071624606847763,
        "stage2_mask_spatial_var_mean": 0.06341055035591125
      },
      {
        "step": 13,
        "loss": 3.6648471355438232,
        "flow_mse": 3.5568251609802246,
        "art_l2": 0.108022041618824,
        "t_mean": 0.6072238683700562,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14481374621391296,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.2039666324853897,
        "stage2_mask_spatial_var_mean": 0.07484860718250275
      },
      {
        "step": 14,
        "loss": 1.8587465286254883,
        "flow_mse": 1.68631911277771,
        "art_l2": 0.17242743074893951,
        "t_mean": 0.6395391821861267,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10025566816329956,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.08088136464357376,
        "stage2_mask_spatial_var_mean": 0.061420951038599014
      },
      {
        "step": 15,
        "loss": 1.9104753732681274,
        "flow_mse": 1.7941375970840454,
        "art_l2": 0.11633777618408203,
        "t_mean": 0.43723514676094055,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.1468224972486496,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10455434769392014,
        "stage2_mask_spatial_var_mean": 0.09443356096744537
      },
      {
        "step": 16,
        "loss": 2.586864948272705,
        "flow_mse": 2.550192356109619,
        "art_l2": 0.03667254000902176,
        "t_mean": 0.6621525883674622,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 9.0,
        "stage2_mask_mass_mean": 0.03778471425175667,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10811017453670502,
        "stage2_mask_spatial_var_mean": 0.07324477285146713
      },
      {
        "step": 17,
        "loss": 2.2332377433776855,
        "flow_mse": 2.1623194217681885,
        "art_l2": 0.07091822475194931,
        "t_mean": 0.5240513682365417,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.07505478709936142,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1669202744960785,
        "stage2_mask_spatial_var_mean": 0.04181981086730957
      },
      {
        "step": 18,
        "loss": 1.9071077108383179,
        "flow_mse": 1.8398103713989258,
        "art_l2": 0.06729736179113388,
        "t_mean": 0.5368333458900452,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 7.0,
        "stage2_mask_mass_mean": 0.05539323389530182,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17302262783050537,
        "stage2_mask_spatial_var_mean": 0.03860964998602867
      },
      {
        "step": 19,
        "loss": 2.763655662536621,
        "flow_mse": 2.6962714195251465,
        "art_l2": 0.06738431751728058,
        "t_mean": 0.2161000519990921,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 8.0,
        "stage2_mask_mass_mean": 0.04516679048538208,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1213332787156105,
        "stage2_mask_spatial_var_mean": 0.04097684845328331
      },
      {
        "step": 20,
        "loss": 2.532846450805664,
        "flow_mse": 2.4885003566741943,
        "art_l2": 0.044346172362565994,
        "t_mean": 0.5189581513404846,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13416120409965515,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1729474812746048,
        "stage2_mask_spatial_var_mean": 0.048083532601594925
      },
      {
        "step": 21,
        "loss": 5.953620433807373,
        "flow_mse": 5.783384799957275,
        "art_l2": 0.17023544013500214,
        "t_mean": 0.11379662156105042,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08619430661201477,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1525188833475113,
        "stage2_mask_spatial_var_mean": 0.03136561065912247
      },
      {
        "step": 22,
        "loss": 3.170893907546997,
        "flow_mse": 2.9966745376586914,
        "art_l2": 0.17421944439411163,
        "t_mean": 0.4425402879714966,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08929875493049622,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.110529825091362,
        "stage2_mask_spatial_var_mean": 0.04687997326254845
      },
      {
        "step": 23,
        "loss": 5.712369441986084,
        "flow_mse": 5.570685386657715,
        "art_l2": 0.14168411493301392,
        "t_mean": 0.20127582550048828,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.11102995276451111,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13087493181228638,
        "stage2_mask_spatial_var_mean": 0.04991722106933594
      },
      {
        "step": 24,
        "loss": 4.49974250793457,
        "flow_mse": 4.380911350250244,
        "art_l2": 0.11883101612329483,
        "t_mean": 0.7069639563560486,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13111761212348938,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.14403989911079407,
        "stage2_mask_spatial_var_mean": 0.04339682310819626
      },
      {
        "step": 25,
        "loss": 8.878008842468262,
        "flow_mse": 8.747542381286621,
        "art_l2": 0.13046608865261078,
        "t_mean": 0.6600659489631653,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16581447422504425,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15605172514915466,
        "stage2_mask_spatial_var_mean": 0.060132116079330444
      },
      {
        "step": 26,
        "loss": 4.466308116912842,
        "flow_mse": 4.335089206695557,
        "art_l2": 0.13121908903121948,
        "t_mean": 0.4758807420730591,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10189919918775558,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13613399863243103,
        "stage2_mask_spatial_var_mean": 0.08625788986682892
      },
      {
        "step": 27,
        "loss": 2.0724141597747803,
        "flow_mse": 1.9720052480697632,
        "art_l2": 0.10040899366140366,
        "t_mean": 0.9104493260383606,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.12326516211032867,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.09395356476306915,
        "stage2_mask_spatial_var_mean": 0.06172376871109009
      },
      {
        "step": 28,
        "loss": 1.0188605785369873,
        "flow_mse": 0.9333723187446594,
        "art_l2": 0.08548831939697266,
        "t_mean": 0.33697742223739624,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 3.0,
        "stage2_mask_mass_mean": 0.07137639820575714,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17038846015930176,
        "stage2_mask_spatial_var_mean": 0.035082150250673294
      },
      {
        "step": 29,
        "loss": 3.0618391036987305,
        "flow_mse": 2.9904816150665283,
        "art_l2": 0.07135753333568573,
        "t_mean": 0.324869304895401,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.0885683000087738,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.06510961055755615,
        "stage2_mask_spatial_var_mean": 0.0735899955034256
      },
      {
        "step": 30,
        "loss": 5.441393852233887,
        "flow_mse": 5.301427841186523,
        "art_l2": 0.13996587693691254,
        "t_mean": 0.2462034672498703,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12016070634126663,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13704855740070343,
        "stage2_mask_spatial_var_mean": 0.03335332125425339
      },
      {
        "step": 31,
        "loss": 5.011932849884033,
        "flow_mse": 4.864713668823242,
        "art_l2": 0.1472189575433731,
        "t_mean": 0.3077169954776764,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.13002191483974457,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1404089778661728,
        "stage2_mask_spatial_var_mean": 0.04379277676343918
      },
      {
        "step": 32,
        "loss": 3.4078190326690674,
        "flow_mse": 3.359482526779175,
        "art_l2": 0.0483364500105381,
        "t_mean": 0.2582360506057739,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11443877220153809,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.18197765946388245,
        "stage2_mask_spatial_var_mean": 0.05140909552574158
      },
      {
        "step": 33,
        "loss": 2.082732915878296,
        "flow_mse": 2.0157032012939453,
        "art_l2": 0.06702974438667297,
        "t_mean": 0.5150696039199829,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 7.0,
        "stage2_mask_mass_mean": 0.04705381393432617,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16342011094093323,
        "stage2_mask_spatial_var_mean": 0.027939317747950554
      },
      {
        "step": 34,
        "loss": 2.758678913116455,
        "flow_mse": 2.591404676437378,
        "art_l2": 0.1672743558883667,
        "t_mean": 0.3112204372882843,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.08132457733154297,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.19395869970321655,
        "stage2_mask_spatial_var_mean": 0.025912052020430565
      },
      {
        "step": 35,
        "loss": 3.688659906387329,
        "flow_mse": 3.560950994491577,
        "art_l2": 0.1277090162038803,
        "t_mean": 0.639766275882721,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12819576263427734,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13321955502033234,
        "stage2_mask_spatial_var_mean": 0.09279214590787888
      },
      {
        "step": 36,
        "loss": 2.638449192047119,
        "flow_mse": 2.594532012939453,
        "art_l2": 0.04391717165708542,
        "t_mean": 0.6604555249214172,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11200390756130219,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.19336432218551636,
        "stage2_mask_spatial_var_mean": 0.042206890881061554
      },
      {
        "step": 37,
        "loss": 2.1486449241638184,
        "flow_mse": 1.9475576877593994,
        "art_l2": 0.2010873407125473,
        "t_mean": 0.756874680519104,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.11723886430263519,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.09051080793142319,
        "stage2_mask_spatial_var_mean": 0.06330938637256622
      },
      {
        "step": 38,
        "loss": 2.670039653778076,
        "flow_mse": 2.581005811691284,
        "art_l2": 0.08903374522924423,
        "t_mean": 0.5363696813583374,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08162893354892731,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.06355810165405273,
        "stage2_mask_spatial_var_mean": 0.06627355515956879
      },
      {
        "step": 39,
        "loss": 2.4632554054260254,
        "flow_mse": 2.3920834064483643,
        "art_l2": 0.07117195427417755,
        "t_mean": 0.580784797668457,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12235209345817566,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.14516600966453552,
        "stage2_mask_spatial_var_mean": 0.08031938225030899
      },
      {
        "step": 40,
        "loss": 1.5618197917938232,
        "flow_mse": 1.4997601509094238,
        "art_l2": 0.06205962225794792,
        "t_mean": 0.38077911734580994,
        "stage2_mask_part_count": 10.0,
        "stage2_mask_empty_parts": 3.0,
        "stage2_mask_mass_mean": 0.05222791060805321,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12488780915737152,
        "stage2_mask_spatial_var_mean": 0.019491467624902725
      }
    ]
  },
  {
    "label": "Official full Stage2 spatial adapter",
    "group": "stage2",
    "phase": "official-full",
    "note": "Official-initialized full matrix Stage2 spatial adapter.",
    "rel_dir": "pact_ot_matrix_official_20260521_full_stage2_spatial_adapter",
    "status": "ok",
    "steps": 40,
    "loss": 5.981064796447754,
    "flow_mse": 5.824027061462402,
    "art_l2": 0.15703772008419037,
    "ot_gate_mean": null,
    "ot_edge_delta": null,
    "ot_virtual_mass": null,
    "ot_modules_active": null,
    "stage2_mask_part_count": 6.0,
    "stage2_mask_empty_parts": 1.0,
    "stage2_mask_centroid_spread": 0.11533430963754654,
    "stage2_mask_spatial_var_mean": 0.046354711055755615,
    "stage2_spatial_adapter_l2": 2.1484719094596378e-20,
    "ot_router": {
      "enabled": false,
      "block_selector": "first-third",
      "sinkhorn_iters": 40,
      "epsilon": 0.1,
      "gate_lambda": 2.5,
      "gate_floor": 0.02,
      "edge_lambda": 0.0,
      "edge_gamma": 8.0,
      "edge_source": "mask",
      "virtual_tokens": 0,
      "virtual_mass": 0.0,
      "detach_plan": false,
      "mask_prior_beta": 0.0
    },
    "stage2_spatial_adapter": {
      "enabled": true,
      "hidden_dim": 64
    },
    "config": {
      "stage": "stage2",
      "config": "/data/250010098/PAct/configs/example_config.json",
      "pact_root": "/data/250010098/PAct",
      "dataset_config": "/data/250010098/PAct-Transporter/evaluations/pact_finetune_full_route_20260516/configs/gapartnet32_stage2_dataset.json",
      "data_dir": null,
      "output_dir": "/data/250010098/PAct-Transporter/reports/20260513_pact_transporter_exploration/runs/pact_ot_matrix_official_20260521_full_stage2_spatial_adapter",
      "finetune_ckpt": "/data/250010098/PAct-Transporter/model_cache/PAct000/PAct/slat_arti_flow_model/model.safetensors",
      "synthetic_smoke": false,
      "device": "cuda",
      "seed": 42,
      "batch_size": 1,
      "num_workers": 0,
      "max_steps": 40,
      "lr": 1e-08,
      "optimizer": "sgd",
      "sgd_momentum": 0.0,
      "weight_decay": 0.01,
      "adam_eps": 1e-08,
      "grad_clip": 1.0,
      "lambda_art": 1.0,
      "enable_stage2_spatial_diagnostics": true,
      "enable_stage2_spatial_adapter": true,
      "stage2_spatial_adapter_hidden": 64,
      "sigma_min": 1e-05,
      "t_schedule": "logitNormal",
      "t_mean": 0.0,
      "t_std": 1.0,
      "print_every": 1,
      "save_every": 0,
      "save_final": true,
      "encode_cond_dino": true,
      "dino_model": "dinov2_vitl14_reg",
      "dino_cache_dir": "/data/250010098/.cache/torch",
      "enable_ot_router": false,
      "ot_block_selector": "first-third",
      "ot_sinkhorn_iters": 40,
      "ot_epsilon": 0.1,
      "ot_gate_lambda": 2.5,
      "ot_gate_floor": 0.02,
      "ot_edge_lambda": 0.0,
      "ot_edge_gamma": 8.0,
      "ot_edge_source": "mask",
      "ot_virtual_tokens": 0,
      "ot_virtual_mass": 0.0,
      "ot_detach_plan": false,
      "ot_mask_prior_beta": 0.0
    },
    "logs": [
      {
        "step": 1,
        "loss": 7.40964412689209,
        "flow_mse": 7.2760443687438965,
        "art_l2": 0.13359981775283813,
        "t_mean": 0.5347283482551575,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14974434673786163,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16794748604297638,
        "stage2_mask_spatial_var_mean": 0.04503370076417923,
        "stage2_spatial_adapter_l2": 0.0
      },
      {
        "step": 2,
        "loss": 3.420111656188965,
        "flow_mse": 3.3164515495300293,
        "art_l2": 0.10366011410951614,
        "t_mean": 0.3323965072631836,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.1470051258802414,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.21312789618968964,
        "stage2_mask_spatial_var_mean": 0.052936047315597534,
        "stage2_spatial_adapter_l2": 3.0255971158846687e-24
      },
      {
        "step": 3,
        "loss": 4.469080924987793,
        "flow_mse": 4.312107086181641,
        "art_l2": 0.15697364509105682,
        "t_mean": 0.8121371269226074,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09739469736814499,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15333124995231628,
        "stage2_mask_spatial_var_mean": 0.054026901721954346,
        "stage2_spatial_adapter_l2": 1.5984205739619367e-23
      },
      {
        "step": 4,
        "loss": 7.818053245544434,
        "flow_mse": 7.688683986663818,
        "art_l2": 0.12936925888061523,
        "t_mean": 0.686140239238739,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.14919649064540863,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17030486464500427,
        "stage2_mask_spatial_var_mean": 0.0551006905734539,
        "stage2_spatial_adapter_l2": 1.3035633633609202e-22
      },
      {
        "step": 5,
        "loss": 3.367090940475464,
        "flow_mse": 3.252946138381958,
        "art_l2": 0.11414473503828049,
        "t_mean": 0.728419840335846,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08144631236791611,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12281860411167145,
        "stage2_mask_spatial_var_mean": 0.0692753717303276,
        "stage2_spatial_adapter_l2": 1.7229454232946943e-22
      },
      {
        "step": 6,
        "loss": 2.2954413890838623,
        "flow_mse": 2.2010445594787598,
        "art_l2": 0.09439685195684433,
        "t_mean": 0.6403265595436096,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.044823698699474335,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10919459164142609,
        "stage2_mask_spatial_var_mean": 0.015086439438164234,
        "stage2_spatial_adapter_l2": 1.6606467425811115e-22
      },
      {
        "step": 7,
        "loss": 1.9940977096557617,
        "flow_mse": 1.9255743026733398,
        "art_l2": 0.0685233622789383,
        "t_mean": 0.5700072646141052,
        "stage2_mask_part_count": 12.0,
        "stage2_mask_empty_parts": 6.0,
        "stage2_mask_mass_mean": 0.04541027545928955,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.0890435129404068,
        "stage2_mask_spatial_var_mean": 0.02842908911406994,
        "stage2_spatial_adapter_l2": 1.5774350437257577e-22
      },
      {
        "step": 8,
        "loss": 2.7402610778808594,
        "flow_mse": 2.6755917072296143,
        "art_l2": 0.06466945260763168,
        "t_mean": 0.655311644077301,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09660336375236511,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1617322862148285,
        "stage2_mask_spatial_var_mean": 0.061530254781246185,
        "stage2_spatial_adapter_l2": 2.8582394544266896e-22
      },
      {
        "step": 9,
        "loss": 8.744171142578125,
        "flow_mse": 8.615007400512695,
        "art_l2": 0.12916402518749237,
        "t_mean": 0.6019466519355774,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16745799779891968,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11526110768318176,
        "stage2_mask_spatial_var_mean": 0.06515448540449142,
        "stage2_spatial_adapter_l2": 3.6076237404665246e-22
      },
      {
        "step": 10,
        "loss": 1.8669387102127075,
        "flow_mse": 1.8236451148986816,
        "art_l2": 0.04329359903931618,
        "t_mean": 0.7785599827766418,
        "stage2_mask_part_count": 10.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.05018261820077896,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.135658860206604,
        "stage2_mask_spatial_var_mean": 0.018480516970157623,
        "stage2_spatial_adapter_l2": 3.662435058338729e-22
      },
      {
        "step": 11,
        "loss": 3.2716774940490723,
        "flow_mse": 3.0789954662323,
        "art_l2": 0.19268211722373962,
        "t_mean": 0.6584867835044861,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.11102995276451111,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13087493181228638,
        "stage2_mask_spatial_var_mean": 0.04991722106933594,
        "stage2_spatial_adapter_l2": 6.461330210805197e-22
      },
      {
        "step": 12,
        "loss": 3.0918636322021484,
        "flow_mse": 3.021379232406616,
        "art_l2": 0.07048437744379044,
        "t_mean": 0.2739311456680298,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09934259206056595,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13076920807361603,
        "stage2_mask_spatial_var_mean": 0.07914616167545319,
        "stage2_spatial_adapter_l2": 9.839359834448364e-22
      },
      {
        "step": 13,
        "loss": 3.906956911087036,
        "flow_mse": 3.7900474071502686,
        "art_l2": 0.11690958589315414,
        "t_mean": 0.6072238683700562,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08315072953701019,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1498417854309082,
        "stage2_mask_spatial_var_mean": 0.03243282437324524,
        "stage2_spatial_adapter_l2": 1.0530437328390524e-21
      },
      {
        "step": 14,
        "loss": 1.857979655265808,
        "flow_mse": 1.6845611333847046,
        "art_l2": 0.1734185516834259,
        "t_mean": 0.6395391821861267,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.06902848929166794,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.05083201825618744,
        "stage2_mask_spatial_var_mean": 0.05934283882379532,
        "stage2_spatial_adapter_l2": 1.2199755861130549e-21
      },
      {
        "step": 15,
        "loss": 3.0202736854553223,
        "flow_mse": 2.8517212867736816,
        "art_l2": 0.1685524582862854,
        "t_mean": 0.43723514676094055,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10390797257423401,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11111626029014587,
        "stage2_mask_spatial_var_mean": 0.05304690822958946,
        "stage2_spatial_adapter_l2": 2.1655265319211486e-21
      },
      {
        "step": 16,
        "loss": 3.8922014236450195,
        "flow_mse": 3.705897092819214,
        "art_l2": 0.18630430102348328,
        "t_mean": 0.6621525883674622,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12016070634126663,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13704855740070343,
        "stage2_mask_spatial_var_mean": 0.03335332125425339,
        "stage2_spatial_adapter_l2": 2.5695761903268512e-21
      },
      {
        "step": 17,
        "loss": 2.4709696769714355,
        "flow_mse": 2.424102544784546,
        "art_l2": 0.04686712101101875,
        "t_mean": 0.5240513682365417,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12271732091903687,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15761402249336243,
        "stage2_mask_spatial_var_mean": 0.05832899361848831,
        "stage2_spatial_adapter_l2": 3.322449690192013e-21
      },
      {
        "step": 18,
        "loss": 7.125680446624756,
        "flow_mse": 6.996397018432617,
        "art_l2": 0.12928330898284912,
        "t_mean": 0.5368333458900452,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.036036036908626556,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.16098950803279877,
        "stage2_mask_spatial_var_mean": 0.032539673149585724,
        "stage2_spatial_adapter_l2": 3.527881691686048e-21
      },
      {
        "step": 19,
        "loss": 4.7238287925720215,
        "flow_mse": 4.6226806640625,
        "art_l2": 0.10114829987287521,
        "t_mean": 0.2161000519990921,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10810810327529907,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.18285012245178223,
        "stage2_mask_spatial_var_mean": 0.0355285182595253,
        "stage2_spatial_adapter_l2": 3.1349299051998373e-21
      },
      {
        "step": 20,
        "loss": 7.621533393859863,
        "flow_mse": 7.504850387573242,
        "art_l2": 0.11668302118778229,
        "t_mean": 0.5189581513404846,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11127345263957977,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1711958348751068,
        "stage2_mask_spatial_var_mean": 0.03285317122936249,
        "stage2_spatial_adapter_l2": 3.807623027300258e-21
      },
      {
        "step": 21,
        "loss": 1.8952378034591675,
        "flow_mse": 1.8203189373016357,
        "art_l2": 0.07491885870695114,
        "t_mean": 0.11379662156105042,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.07638527452945709,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1542912721633911,
        "stage2_mask_spatial_var_mean": 0.02149767056107521,
        "stage2_spatial_adapter_l2": 3.3694764024337514e-21
      },
      {
        "step": 22,
        "loss": 3.2186951637268066,
        "flow_mse": 3.0395572185516357,
        "art_l2": 0.1791379451751709,
        "t_mean": 0.4425402879714966,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.15157049894332886,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12493190169334412,
        "stage2_mask_spatial_var_mean": 0.07618506997823715,
        "stage2_spatial_adapter_l2": 3.63830020604229e-21
      },
      {
        "step": 23,
        "loss": 4.5477423667907715,
        "flow_mse": 4.419948101043701,
        "art_l2": 0.1277943104505539,
        "t_mean": 0.20127582550048828,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.08777697384357452,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15772874653339386,
        "stage2_mask_spatial_var_mean": 0.0357699990272522,
        "stage2_spatial_adapter_l2": 4.2455650784459355e-21
      },
      {
        "step": 24,
        "loss": 1.9257031679153442,
        "flow_mse": 1.7393343448638916,
        "art_l2": 0.18636885285377502,
        "t_mean": 0.7069639563560486,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.07651570439338684,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.06106841191649437,
        "stage2_mask_spatial_var_mean": 0.05943670868873596,
        "stage2_spatial_adapter_l2": 5.1611118448743546e-21
      },
      {
        "step": 25,
        "loss": 8.639487266540527,
        "flow_mse": 8.508219718933105,
        "art_l2": 0.13126729428768158,
        "t_mean": 0.6600659489631653,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16581445932388306,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15902292728424072,
        "stage2_mask_spatial_var_mean": 0.06032077595591545,
        "stage2_spatial_adapter_l2": 7.069259190495068e-21
      },
      {
        "step": 26,
        "loss": 6.612235069274902,
        "flow_mse": 6.570220470428467,
        "art_l2": 0.04201442748308182,
        "t_mean": 0.4758807420730591,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.18504993617534637,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.15875029563903809,
        "stage2_mask_spatial_var_mean": 0.05624683201313019,
        "stage2_spatial_adapter_l2": 8.180166792121619e-21
      },
      {
        "step": 27,
        "loss": 3.420578956604004,
        "flow_mse": 3.2957637310028076,
        "art_l2": 0.1248152107000351,
        "t_mean": 0.9104493260383606,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09252496808767319,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.142442524433136,
        "stage2_mask_spatial_var_mean": 0.02995327115058899,
        "stage2_spatial_adapter_l2": 7.408824102090188e-21
      },
      {
        "step": 28,
        "loss": 4.4891157150268555,
        "flow_mse": 4.337026119232178,
        "art_l2": 0.15208977460861206,
        "t_mean": 0.33697742223739624,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.15029218792915344,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.210830956697464,
        "stage2_mask_spatial_var_mean": 0.05848394334316254,
        "stage2_spatial_adapter_l2": 9.028317725672573e-21
      },
      {
        "step": 29,
        "loss": 1.0369130373001099,
        "flow_mse": 0.9547207951545715,
        "art_l2": 0.08219219744205475,
        "t_mean": 0.324869304895401,
        "stage2_mask_part_count": 7.0,
        "stage2_mask_empty_parts": 3.0,
        "stage2_mask_mass_mean": 0.07137639820575714,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.17038846015930176,
        "stage2_mask_spatial_var_mean": 0.035082150250673294,
        "stage2_spatial_adapter_l2": 1.0091845811217022e-20
      },
      {
        "step": 30,
        "loss": 3.3686954975128174,
        "flow_mse": 3.306339740753174,
        "art_l2": 0.06235581263899803,
        "t_mean": 0.2462034672498703,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12198685109615326,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10257704555988312,
        "stage2_mask_spatial_var_mean": 0.06817074865102768,
        "stage2_spatial_adapter_l2": 1.0567400346426853e-20
      },
      {
        "step": 31,
        "loss": 3.152669906616211,
        "flow_mse": 3.0884249210357666,
        "art_l2": 0.06424497067928314,
        "t_mean": 0.3077169954776764,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.09495982527732849,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.0754053071141243,
        "stage2_mask_spatial_var_mean": 0.07356518507003784,
        "stage2_spatial_adapter_l2": 1.2294500151062157e-20
      },
      {
        "step": 32,
        "loss": 3.18445086479187,
        "flow_mse": 3.071899652481079,
        "art_l2": 0.11255116760730743,
        "t_mean": 0.2582360506057739,
        "stage2_mask_part_count": 11.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.05717511847615242,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.14053958654403687,
        "stage2_mask_spatial_var_mean": 0.02332073263823986,
        "stage2_spatial_adapter_l2": 1.043602483966654e-20
      },
      {
        "step": 33,
        "loss": 2.939969301223755,
        "flow_mse": 2.7102608680725098,
        "art_l2": 0.22970837354660034,
        "t_mean": 0.5150696039199829,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.12417824566364288,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10249984264373779,
        "stage2_mask_spatial_var_mean": 0.06571410596370697,
        "stage2_spatial_adapter_l2": 1.1873594466243377e-20
      },
      {
        "step": 34,
        "loss": 5.8472371101379395,
        "flow_mse": 5.695064067840576,
        "art_l2": 0.15217319130897522,
        "t_mean": 0.3112204372882843,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.10153397172689438,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.12184513360261917,
        "stage2_mask_spatial_var_mean": 0.06188197433948517,
        "stage2_spatial_adapter_l2": 1.343011169555306e-20
      },
      {
        "step": 35,
        "loss": 3.002622365951538,
        "flow_mse": 2.802635669708252,
        "art_l2": 0.19998668134212494,
        "t_mean": 0.639766275882721,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.16995374858379364,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.09016727656126022,
        "stage2_mask_spatial_var_mean": 0.07749043405056,
        "stage2_spatial_adapter_l2": 1.4286029564250203e-20
      },
      {
        "step": 36,
        "loss": 2.8920681476593018,
        "flow_mse": 2.7428979873657227,
        "art_l2": 0.1491701602935791,
        "t_mean": 0.6604555249214172,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10007305443286896,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.13893088698387146,
        "stage2_mask_spatial_var_mean": 0.03599216043949127,
        "stage2_spatial_adapter_l2": 1.70272996519024e-20
      },
      {
        "step": 37,
        "loss": 1.799229383468628,
        "flow_mse": 1.6950461864471436,
        "art_l2": 0.10418317466974258,
        "t_mean": 0.756874680519104,
        "stage2_mask_part_count": 4.0,
        "stage2_mask_empty_parts": 2.0,
        "stage2_mask_mass_mean": 0.1468224972486496,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10455434769392014,
        "stage2_mask_spatial_var_mean": 0.09443356096744537,
        "stage2_spatial_adapter_l2": 2.0167671460414364e-20
      },
      {
        "step": 38,
        "loss": 2.981696367263794,
        "flow_mse": 2.8329010009765625,
        "art_l2": 0.14879527688026428,
        "t_mean": 0.5363696813583374,
        "stage2_mask_part_count": 3.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.11541271209716797,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.10026109218597412,
        "stage2_mask_spatial_var_mean": 0.04383675381541252,
        "stage2_spatial_adapter_l2": 2.1629881214163765e-20
      },
      {
        "step": 39,
        "loss": 2.542867660522461,
        "flow_mse": 2.4711506366729736,
        "art_l2": 0.07171696424484253,
        "t_mean": 0.580784797668457,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.09435111284255981,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.1347745954990387,
        "stage2_mask_spatial_var_mean": 0.030499106273055077,
        "stage2_spatial_adapter_l2": 2.088441861443016e-20
      },
      {
        "step": 40,
        "loss": 5.981064796447754,
        "flow_mse": 5.824027061462402,
        "art_l2": 0.15703772008419037,
        "t_mean": 0.38077911734580994,
        "stage2_mask_part_count": 6.0,
        "stage2_mask_empty_parts": 1.0,
        "stage2_mask_mass_mean": 0.10311663150787354,
        "stage2_mask_mass_min": 0.0,
        "stage2_mask_centroid_spread": 0.11533430963754654,
        "stage2_mask_spatial_var_mean": 0.046354711055755615,
        "stage2_spatial_adapter_l2": 2.1484719094596378e-20
      }
    ]
  }
]