{
  "samples": {
    "trajectory_count": 103,
    "loaded_frame_count": 25328,
    "step_stride": 1,
    "tasks": [
      "lift_cube",
      "pick_place_bread",
      "pick_place_can",
      "pick_place_cereal",
      "pick_place_milk",
      "stack_blocks"
    ],
    "difficulties": [
      "low",
      "medium"
    ]
  },
  "epochs": 40,
  "batch_size": 128,
  "learning_rate": 0.0003,
  "weight_decay": 0.0001,
  "device": "cuda",
  "history": [
    {
      "epoch": 1.0,
      "train_mse": 0.053797176164960755,
      "val_mse": 0.02619670540988916
    },
    {
      "epoch": 2.0,
      "train_mse": 0.026044019152300674,
      "val_mse": 0.02725875084407172
    },
    {
      "epoch": 3.0,
      "train_mse": 0.025213642344257877,
      "val_mse": 0.029101956754257006
    },
    {
      "epoch": 4.0,
      "train_mse": 0.022870065035526597,
      "val_mse": 0.026456370199810866
    },
    {
      "epoch": 5.0,
      "train_mse": 0.02211590029456468,
      "val_mse": 0.020979402889061447
    },
    {
      "epoch": 6.0,
      "train_mse": 0.02168502898569441,
      "val_mse": 0.019284333582253736
    },
    {
      "epoch": 7.0,
      "train_mse": 0.02108741015167085,
      "val_mse": 0.022391801277527393
    },
    {
      "epoch": 8.0,
      "train_mse": 0.02143752471068137,
      "val_mse": 0.020186543975826636
    },
    {
      "epoch": 9.0,
      "train_mse": 0.020406250103426976,
      "val_mse": 0.022699372866868812
    },
    {
      "epoch": 10.0,
      "train_mse": 0.020862830311643682,
      "val_mse": 0.020913607755113042
    },
    {
      "epoch": 11.0,
      "train_mse": 0.020367756763015256,
      "val_mse": 0.019451317252329365
    },
    {
      "epoch": 12.0,
      "train_mse": 0.020526346440450212,
      "val_mse": 0.019641175331420485
    },
    {
      "epoch": 13.0,
      "train_mse": 0.01967800536257218,
      "val_mse": 0.018605292394935507
    },
    {
      "epoch": 14.0,
      "train_mse": 0.019654701238197073,
      "val_mse": 0.019030479458271046
    },
    {
      "epoch": 15.0,
      "train_mse": 0.02077592885713932,
      "val_mse": 0.019544998643307353
    },
    {
      "epoch": 16.0,
      "train_mse": 0.019671596932293728,
      "val_mse": 0.018935938501325748
    },
    {
      "epoch": 17.0,
      "train_mse": 0.019606111569670907,
      "val_mse": 0.019572921031701378
    },
    {
      "epoch": 18.0,
      "train_mse": 0.01944679094589286,
      "val_mse": 0.018906898973112422
    },
    {
      "epoch": 19.0,
      "train_mse": 0.019540326114760944,
      "val_mse": 0.01921737950048875
    },
    {
      "epoch": 20.0,
      "train_mse": 0.019115112357220777,
      "val_mse": 0.018009563542682197
    },
    {
      "epoch": 21.0,
      "train_mse": 0.019305372959566483,
      "val_mse": 0.019896589211162826
    },
    {
      "epoch": 22.0,
      "train_mse": 0.020241556941134654,
      "val_mse": 0.018675044545522528
    },
    {
      "epoch": 23.0,
      "train_mse": 0.01905667615024388,
      "val_mse": 0.023118399357693484
    },
    {
      "epoch": 24.0,
      "train_mse": 0.01883897730293954,
      "val_mse": 0.01874487928977214
    },
    {
      "epoch": 25.0,
      "train_mse": 0.0190682219243387,
      "val_mse": 0.01918733736054135
    },
    {
      "epoch": 26.0,
      "train_mse": 0.01879240737178489,
      "val_mse": 0.020210167323727823
    },
    {
      "epoch": 27.0,
      "train_mse": 0.018606472403369288,
      "val_mse": 0.017827463860276106
    },
    {
      "epoch": 28.0,
      "train_mse": 0.019354441514080854,
      "val_mse": 0.01885381301297907
    },
    {
      "epoch": 29.0,
      "train_mse": 0.019082575099737858,
      "val_mse": 0.01816060308462017
    },
    {
      "epoch": 30.0,
      "train_mse": 0.018576878484549007,
      "val_mse": 0.017810351179781952
    },
    {
      "epoch": 31.0,
      "train_mse": 0.01863790871773211,
      "val_mse": 0.017086478529060378
    },
    {
      "epoch": 32.0,
      "train_mse": 0.018316452185579063,
      "val_mse": 0.01812759830730544
    },
    {
      "epoch": 33.0,
      "train_mse": 0.018366531492950394,
      "val_mse": 0.017392945919153087
    },
    {
      "epoch": 34.0,
      "train_mse": 0.01844127951175168,
      "val_mse": 0.016948535839985627
    },
    {
      "epoch": 35.0,
      "train_mse": 0.018191324602822332,
      "val_mse": 0.01724688478421207
    },
    {
      "epoch": 36.0,
      "train_mse": 0.017863313722926148,
      "val_mse": 0.018118728501240897
    },
    {
      "epoch": 37.0,
      "train_mse": 0.018502490333533777,
      "val_mse": 0.01816481517308777
    },
    {
      "epoch": 38.0,
      "train_mse": 0.018186402516808885,
      "val_mse": 0.021513568765824925
    },
    {
      "epoch": 39.0,
      "train_mse": 0.018155591582981072,
      "val_mse": 0.01769615077886321
    },
    {
      "epoch": 40.0,
      "train_mse": 0.01812755108300605,
      "val_mse": 0.018309524489213157
    }
  ],
  "best_val_mse": 0.016948535839985627,
  "global_steps": 7160,
  "output_files": [
    "flock_robotics_adapter.py",
    "model.pt",
    "vla_config.json",
    "training_report.json"
  ]
}