{
  "model_parameters": 521731,
  "device": "cuda",
  "train_samples": 180900,
  "validation_samples": 36180,
  "scenario_counts": {
    "0": {
      "slide_support_and_unload": 11400,
      "unsupported_unload": 2880,
      "one_hand_blocked_tray": 3300,
      "two_hand_tray": 12300,
      "free_motion": 6300
    },
    "1": {
      "slide_support_and_unload": 11400,
      "unsupported_unload": 2880,
      "one_hand_blocked_tray": 3300,
      "two_hand_tray": 12300,
      "free_motion": 6300
    },
    "2": {
      "slide_support_and_unload": 11400,
      "unsupported_unload": 2880,
      "one_hand_blocked_tray": 3300,
      "two_hand_tray": 12300,
      "free_motion": 6300
    },
    "3": {
      "slide_support_and_unload": 11400,
      "unsupported_unload": 2880,
      "one_hand_blocked_tray": 3300,
      "two_hand_tray": 12300,
      "free_motion": 6300
    },
    "4": {
      "slide_support_and_unload": 11400,
      "unsupported_unload": 2880,
      "one_hand_blocked_tray": 3300,
      "two_hand_tray": 12300,
      "free_motion": 6300
    },
    "5": {
      "slide_support_and_unload": 11400,
      "unsupported_unload": 2880,
      "one_hand_blocked_tray": 3300,
      "two_hand_tray": 12300,
      "free_motion": 6300
    }
  },
  "epochs": [
    {
      "epoch": 1,
      "train_mse": 0.11669006642330898,
      "val_mse": 0.08554307826691204
    },
    {
      "epoch": 2,
      "train_mse": 0.0691971659158053,
      "val_mse": 0.04707959211534924
    },
    {
      "epoch": 3,
      "train_mse": 0.029611097209239275,
      "val_mse": 0.020102285107390747
    },
    {
      "epoch": 4,
      "train_mse": 0.016996806864155813,
      "val_mse": 0.015492904310425123
    },
    {
      "epoch": 5,
      "train_mse": 0.013970357742620988,
      "val_mse": 0.013470031398658952
    },
    {
      "epoch": 6,
      "train_mse": 0.012361691532175192,
      "val_mse": 0.012300785889642106
    },
    {
      "epoch": 7,
      "train_mse": 0.011182273089299711,
      "val_mse": 0.01122638521095117
    },
    {
      "epoch": 8,
      "train_mse": 0.010202664798230267,
      "val_mse": 0.010520024147505561
    },
    {
      "epoch": 9,
      "train_mse": 0.0094479773235455,
      "val_mse": 0.009796942946397595
    },
    {
      "epoch": 10,
      "train_mse": 0.008868326699842563,
      "val_mse": 0.009493966974938909
    },
    {
      "epoch": 11,
      "train_mse": 0.00839696324357156,
      "val_mse": 0.008874511372091042
    },
    {
      "epoch": 12,
      "train_mse": 0.007969921821037704,
      "val_mse": 0.008672614853518704
    },
    {
      "epoch": 13,
      "train_mse": 0.007631532523488061,
      "val_mse": 0.008500015557122728
    },
    {
      "epoch": 14,
      "train_mse": 0.007289170397424631,
      "val_mse": 0.007943349166048897
    },
    {
      "epoch": 15,
      "train_mse": 0.0069637455967035185,
      "val_mse": 0.007678921024004619
    },
    {
      "epoch": 16,
      "train_mse": 0.006713732470119937,
      "val_mse": 0.007405612384900451
    },
    {
      "epoch": 17,
      "train_mse": 0.006507526763997386,
      "val_mse": 0.007233798077019553
    },
    {
      "epoch": 18,
      "train_mse": 0.006346743459698189,
      "val_mse": 0.007071815211222404
    },
    {
      "epoch": 19,
      "train_mse": 0.006116720915803414,
      "val_mse": 0.00670566057993306
    },
    {
      "epoch": 20,
      "train_mse": 0.005908547947706466,
      "val_mse": 0.0066809831704530455
    }
  ],
  "wall_s": 53.361429929733276,
  "checkpoint": "/root/autodl-tmp/third-hand/artifacts/policy/contact_mlp.pt"
}