[
  {
    "method": "flowmo",
    "steps": 20000,
    "batch_size": 256,
    "train_samples": 5120000,
    "final_train_loss": 0.004372047260403633,
    "total_parameters": 663964,
    "target_mode": "absolute_normalized",
    "position_scale": 5.0,
    "heading_weight": 2.0,
    "current_pose_weight": 1.0,
    "motion_weight": 0.5,
    "precision": "bf16",
    "checkpoint_name": "paper.pt",
    "final_checkpoint": "paper.pt",
    "intermediate_checkpoints": [
      "paper_step_002000.pt",
      "paper_step_004000.pt",
      "paper_step_006000.pt",
      "paper_step_008000.pt",
      "paper_step_010000.pt",
      "paper_step_012000.pt",
      "paper_step_014000.pt",
      "paper_step_016000.pt",
      "paper_step_018000.pt",
      "paper_step_020000.pt"
    ],
    "checkpoint_interval": 2000,
    "prediction": {
      "pos1": 0.07179122391001631,
      "heading1": 0.045128354569897056,
      "pos3": 0.07811223280926545,
      "heading3": 0.04785821299689511,
      "pos6": 0.08915455282355349,
      "heading6": 0.05167978972895071,
      "pos8": 0.09605406736955047,
      "heading8": 0.054166459323217474,
      "pos10": 0.10336457837062578,
      "heading10": 0.05640091137805333,
      "pos20": 0.1460142444508771,
      "heading20": 0.06746077448284875
    }
  },
  {
    "method": "leworldmodel",
    "steps": 20000,
    "batch_size": 256,
    "train_samples": 5120000,
    "final_train_loss": 0.018198927864432335,
    "total_parameters": 664612,
    "target_mode": "absolute_normalized",
    "position_scale": 5.0,
    "heading_weight": 2.0,
    "current_pose_weight": 1.0,
    "motion_weight": 0.5,
    "precision": "bf16",
    "checkpoint_name": "paper.pt",
    "final_checkpoint": "paper.pt",
    "intermediate_checkpoints": [
      "paper_step_002000.pt",
      "paper_step_004000.pt",
      "paper_step_006000.pt",
      "paper_step_008000.pt",
      "paper_step_010000.pt",
      "paper_step_012000.pt",
      "paper_step_014000.pt",
      "paper_step_016000.pt",
      "paper_step_018000.pt",
      "paper_step_020000.pt"
    ],
    "checkpoint_interval": 2000,
    "prediction": {
      "pos1": 0.10107660254773994,
      "heading1": 0.0479962204505379,
      "pos3": 0.11733688049328823,
      "heading3": 0.054719595277371504,
      "pos6": 0.14297669551645717,
      "heading6": 0.0649993756475548,
      "pos8": 0.1607180543554326,
      "heading8": 0.07113518511566023,
      "pos10": 0.1783415130339563,
      "heading10": 0.07669965278667708,
      "pos20": 0.25807287978629273,
      "heading20": 0.0995667139844348
    }
  },
  {
    "method": "planet",
    "steps": 20000,
    "batch_size": 256,
    "train_samples": 5120000,
    "final_train_loss": 0.008471474051475525,
    "total_parameters": 664644,
    "target_mode": "absolute_normalized",
    "position_scale": 5.0,
    "heading_weight": 2.0,
    "current_pose_weight": 1.0,
    "motion_weight": 0.5,
    "precision": "bf16",
    "checkpoint_name": "paper.pt",
    "final_checkpoint": "paper.pt",
    "intermediate_checkpoints": [
      "paper_step_002000.pt",
      "paper_step_004000.pt",
      "paper_step_006000.pt",
      "paper_step_008000.pt",
      "paper_step_010000.pt",
      "paper_step_012000.pt",
      "paper_step_014000.pt",
      "paper_step_016000.pt",
      "paper_step_018000.pt",
      "paper_step_020000.pt"
    ],
    "checkpoint_interval": 2000,
    "prediction": {
      "pos1": 0.12035437487065792,
      "heading1": 0.050536025839392096,
      "pos3": 0.12147162947803736,
      "heading3": 0.05107929309209188,
      "pos6": 0.13511362741701305,
      "heading6": 0.05543786101043224,
      "pos8": 0.14701085817068815,
      "heading8": 0.05840683872035394,
      "pos10": 0.16024279454723,
      "heading10": 0.061416062798040606,
      "pos20": 0.22345823872213563,
      "heading20": 0.07620794516211997
    }
  },
  {
    "method": "tdmpc2",
    "steps": 20000,
    "batch_size": 256,
    "train_samples": 5120000,
    "final_train_loss": 0.010143937543034554,
    "total_parameters": 667780,
    "target_mode": "absolute_normalized",
    "position_scale": 5.0,
    "heading_weight": 2.0,
    "current_pose_weight": 1.0,
    "motion_weight": 0.5,
    "precision": "bf16",
    "checkpoint_name": "paper.pt",
    "final_checkpoint": "paper.pt",
    "intermediate_checkpoints": [
      "paper_step_002000.pt",
      "paper_step_004000.pt",
      "paper_step_006000.pt",
      "paper_step_008000.pt",
      "paper_step_010000.pt",
      "paper_step_012000.pt",
      "paper_step_014000.pt",
      "paper_step_016000.pt",
      "paper_step_018000.pt",
      "paper_step_020000.pt"
    ],
    "checkpoint_interval": 2000,
    "prediction": {
      "pos1": 0.09520710981450975,
      "heading1": 0.048679903343630336,
      "pos3": 0.10856497256706159,
      "heading3": 0.053618756433327995,
      "pos6": 0.1308438451960683,
      "heading6": 0.06132081007429709,
      "pos8": 0.14709665269280472,
      "heading8": 0.06618776960143198,
      "pos10": 0.1638579194744428,
      "heading10": 0.07064890124214192,
      "pos20": 0.23970605929692587,
      "heading20": 0.08867141495769222
    }
  }
]