{
  "checkpoint": "outputs/r4_small_8L_t8/checkpoint-150000/checkpoint.pt",
  "format": "edgevla",
  "train_args": {
    "fastvit_variant": "fastvit_t8",
    "fastvit_pretrained": true,
    "fastvit_input_size": 256,
    "freeze_fastvit_backbone": false,
    "base_policy": "lerobot/smolvla_base",
    "dataset": "lerobot/fmb",
    "rename_map": null,
    "run_name": "r4_small_8L_t8",
    "output_dir": "outputs",
    "batch_size": 64,
    "steps": 150000,
    "lr": 0.0001,
    "weight_decay": 1e-10,
    "warmup_steps": 2000,
    "grad_clip_norm": 10.0,
    "num_workers": 7,
    "log_freq": 50,
    "save_freq": 10000,
    "seed": 42,
    "train_expert_only": true,
    "empty_cameras": 0,
    "num_vlm_layers": 8,
    "num_expert_layers": -1,
    "expert_width_multiplier": 0.75,
    "image_aug": true,
    "resume": "outputs/r3_small_8L_t8/checkpoint-50000/checkpoint.pt"
  },
  "step": 150000,
  "train_loss": 0.0807022899389267,
  "model_size": {
    "total_params": 228032560,
    "total_params_M": 228.0,
    "trainable_params": 54755440,
    "trainable_params_M": 54.8,
    "memory_fp32_MB": 869.9,
    "memory_fp16_MB": 434.9
  },
  "flow_matching_loss": 0.07548771565780044,
  "action_metrics": {
    "action_mse": 0.5150853395462036,
    "action_mae": 0.34801334142684937,
    "action_rmse": 0.7176944611366342,
    "cosine_similarity_mean": 0.6786063313484192,
    "cosine_similarity_std": 0.1838311403989792,
    "mse/x": 0.5143061876296997,
    "mae/x": 0.39246663451194763,
    "mse/y": 0.5399093627929688,
    "mae/y": 0.3091602325439453,
    "mse/z": 0.5292629599571228,
    "mae/z": 0.42114609479904175,
    "mse/rx": 0.46191930770874023,
    "mae/rx": 0.3582588732242584,
    "mse/ry": 0.9737266898155212,
    "mae/ry": 0.5462997555732727,
    "mse/rz": 0.3945960998535156,
    "mae/rz": 0.3141518831253052,
    "mse/gripper": 0.1918761283159256,
    "mae/gripper": 0.09460986405611038,
    "mse_horizon_1": 0.4286166727542877,
    "mse_horizon_5": 0.37619855999946594,
    "mse_horizon_10": 0.3926026225090027,
    "mse_horizon_25": 0.48308753967285156,
    "mse_horizon_50": 0.5150853395462036,
    "gripper_binary_accuracy": 0.9583600163459778,
    "pred_jerk_magnitude": 0.4423182010650635,
    "gt_jerk_magnitude": 0.2543262839317322,
    "jerk_ratio": 1.7391761253579014
  },
  "latency": {
    "latency_mean_ms": 90.09251759620383,
    "latency_std_ms": 14.897770921482905,
    "latency_p50_ms": 86.886189004872,
    "latency_p95_ms": 105.17390802269797,
    "latency_p99_ms": 156.08970184752243,
    "throughput_hz": 11.099700915030665
  }
}