| [ |
| { |
| "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 |
| } |
| } |
| ] |
