diff --git a/.gitattributes b/.gitattributes
index a6344aac8c09253b3b630fb776ae94478aa0275b..d71f7146243d39d3fc6e82025b50a8c1e3759b08 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -33,3 +33,43 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_zigzag_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_station_keeping_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_twin_passive_to_active_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_waypoint_zigzag_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_waypoint_zigzag_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_reach_uniform_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_square_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_square_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_zigzag_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_twin_waypoint_zigzag_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_waypoint_square_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_reach_uniform_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_square_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_zigzag_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_twin_waypoint_zigzag_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_zigzag_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_passive_to_active_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_zigzag_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_passive_to_active_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_passive_to_active_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_reach_uniform_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_passive_to_active_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_counterflow_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_waypoint_zigzag_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_current_estimator_mpc_inferred_triangle_waypoint_zigzag_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep000.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep002.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep001.gif filter=lfs diff=lfs merge=lfs -text
+experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_square_ep001.gif filter=lfs diff=lfs merge=lfs -text
diff --git a/experiments/flowmo/result/paper_training.json b/experiments/flowmo/result/paper_training.json
new file mode 100644
index 0000000000000000000000000000000000000000..3c29c5111ed159eed838b822015640c13542cc41
--- /dev/null
+++ b/experiments/flowmo/result/paper_training.json
@@ -0,0 +1,43 @@
+{
+  "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
+  }
+}
\ No newline at end of file
diff --git a/experiments/flowmo/result/paper_training_trace.jsonl b/experiments/flowmo/result/paper_training_trace.jsonl
new file mode 100644
index 0000000000000000000000000000000000000000..bf0d44c0b89fe9e160c27b01f2ab3614156c5f55
--- /dev/null
+++ b/experiments/flowmo/result/paper_training_trace.jsonl
@@ -0,0 +1,100 @@
+{"method": "flowmo", "step": 200, "loss": 2.5686612129211426}
+{"method": "flowmo", "step": 400, "loss": 2.5599124431610107}
+{"method": "flowmo", "step": 600, "loss": 2.1556336879730225}
+{"method": "flowmo", "step": 800, "loss": 1.8842406272888184}
+{"method": "flowmo", "step": 1000, "loss": 1.8033578395843506}
+{"method": "flowmo", "step": 1200, "loss": 1.6628687381744385}
+{"method": "flowmo", "step": 1400, "loss": 1.5624256134033203}
+{"method": "flowmo", "step": 1600, "loss": 1.4740279912948608}
+{"method": "flowmo", "step": 1800, "loss": 1.3885352611541748}
+{"method": "flowmo", "step": 2000, "loss": 1.324019193649292}
+{"method": "flowmo", "step": 2200, "loss": 1.0553849935531616}
+{"method": "flowmo", "step": 2400, "loss": 0.6030915975570679}
+{"method": "flowmo", "step": 2600, "loss": 0.20210689306259155}
+{"method": "flowmo", "step": 2800, "loss": 0.14605776965618134}
+{"method": "flowmo", "step": 3000, "loss": 0.09821392595767975}
+{"method": "flowmo", "step": 3200, "loss": 0.07479490339756012}
+{"method": "flowmo", "step": 3400, "loss": 0.059080999344587326}
+{"method": "flowmo", "step": 3600, "loss": 0.051076825708150864}
+{"method": "flowmo", "step": 3800, "loss": 0.042308710515499115}
+{"method": "flowmo", "step": 4000, "loss": 0.040146660059690475}
+{"method": "flowmo", "step": 4200, "loss": 0.03381121903657913}
+{"method": "flowmo", "step": 4400, "loss": 0.033231284469366074}
+{"method": "flowmo", "step": 4600, "loss": 0.028457675129175186}
+{"method": "flowmo", "step": 4800, "loss": 0.029077233746647835}
+{"method": "flowmo", "step": 5000, "loss": 0.02207356132566929}
+{"method": "flowmo", "step": 5200, "loss": 0.020034978166222572}
+{"method": "flowmo", "step": 5400, "loss": 0.019785162061452866}
+{"method": "flowmo", "step": 5600, "loss": 0.018391719087958336}
+{"method": "flowmo", "step": 5800, "loss": 0.02175654098391533}
+{"method": "flowmo", "step": 6000, "loss": 0.015171783976256847}
+{"method": "flowmo", "step": 6200, "loss": 0.01452728919684887}
+{"method": "flowmo", "step": 6400, "loss": 0.013214356265962124}
+{"method": "flowmo", "step": 6600, "loss": 0.051673468202352524}
+{"method": "flowmo", "step": 6800, "loss": 0.018827352672815323}
+{"method": "flowmo", "step": 7000, "loss": 0.012735347263514996}
+{"method": "flowmo", "step": 7200, "loss": 0.011451991274952888}
+{"method": "flowmo", "step": 7400, "loss": 0.010433687828481197}
+{"method": "flowmo", "step": 7600, "loss": 0.010923548601567745}
+{"method": "flowmo", "step": 7800, "loss": 0.010971073061227798}
+{"method": "flowmo", "step": 8000, "loss": 0.009853748604655266}
+{"method": "flowmo", "step": 8200, "loss": 0.09088479727506638}
+{"method": "flowmo", "step": 8400, "loss": 0.034223418682813644}
+{"method": "flowmo", "step": 8600, "loss": 0.014456425793468952}
+{"method": "flowmo", "step": 8800, "loss": 0.009422067552804947}
+{"method": "flowmo", "step": 9000, "loss": 0.00858109537512064}
+{"method": "flowmo", "step": 9200, "loss": 0.00857796985656023}
+{"method": "flowmo", "step": 9400, "loss": 0.008296442218124866}
+{"method": "flowmo", "step": 9600, "loss": 0.008247998543083668}
+{"method": "flowmo", "step": 9800, "loss": 0.008240980096161366}
+{"method": "flowmo", "step": 10000, "loss": 0.008153271861374378}
+{"method": "flowmo", "step": 10200, "loss": 0.012404488399624825}
+{"method": "flowmo", "step": 10400, "loss": 0.013864593580365181}
+{"method": "flowmo", "step": 10600, "loss": 0.01010044477880001}
+{"method": "flowmo", "step": 10800, "loss": 0.00767604261636734}
+{"method": "flowmo", "step": 11000, "loss": 0.007007307838648558}
+{"method": "flowmo", "step": 11200, "loss": 0.0070138657465577126}
+{"method": "flowmo", "step": 11400, "loss": 0.007243836764246225}
+{"method": "flowmo", "step": 11600, "loss": 0.006900576408952475}
+{"method": "flowmo", "step": 11800, "loss": 0.0068667978048324585}
+{"method": "flowmo", "step": 12000, "loss": 0.006599605083465576}
+{"method": "flowmo", "step": 12200, "loss": 0.007158435881137848}
+{"method": "flowmo", "step": 12400, "loss": 0.045721929520368576}
+{"method": "flowmo", "step": 12600, "loss": 0.006790271960198879}
+{"method": "flowmo", "step": 12800, "loss": 0.0060927667655050755}
+{"method": "flowmo", "step": 13000, "loss": 0.005786360241472721}
+{"method": "flowmo", "step": 13200, "loss": 0.00602421211078763}
+{"method": "flowmo", "step": 13400, "loss": 0.005942641757428646}
+{"method": "flowmo", "step": 13600, "loss": 0.006074435543268919}
+{"method": "flowmo", "step": 13800, "loss": 0.021174009889364243}
+{"method": "flowmo", "step": 14000, "loss": 0.006621338427066803}
+{"method": "flowmo", "step": 14200, "loss": 0.005491666030138731}
+{"method": "flowmo", "step": 14400, "loss": 0.0051383040845394135}
+{"method": "flowmo", "step": 14600, "loss": 0.005117133259773254}
+{"method": "flowmo", "step": 14800, "loss": 0.0053353263065218925}
+{"method": "flowmo", "step": 15000, "loss": 0.00533561734482646}
+{"method": "flowmo", "step": 15200, "loss": 0.005121554713696241}
+{"method": "flowmo", "step": 15400, "loss": 0.005291329696774483}
+{"method": "flowmo", "step": 15600, "loss": 0.00511613953858614}
+{"method": "flowmo", "step": 15800, "loss": 0.005213129799813032}
+{"method": "flowmo", "step": 16000, "loss": 0.005071689374744892}
+{"method": "flowmo", "step": 16200, "loss": 0.0057407282292842865}
+{"method": "flowmo", "step": 16400, "loss": 0.0054640620946884155}
+{"method": "flowmo", "step": 16600, "loss": 0.005037755239754915}
+{"method": "flowmo", "step": 16800, "loss": 0.004956530407071114}
+{"method": "flowmo", "step": 17000, "loss": 0.24728184938430786}
+{"method": "flowmo", "step": 17200, "loss": 0.0358046218752861}
+{"method": "flowmo", "step": 17400, "loss": 0.005404628813266754}
+{"method": "flowmo", "step": 17600, "loss": 0.004861537832766771}
+{"method": "flowmo", "step": 17800, "loss": 0.0046697030775249004}
+{"method": "flowmo", "step": 18000, "loss": 0.00479076337069273}
+{"method": "flowmo", "step": 18200, "loss": 0.0045429919846355915}
+{"method": "flowmo", "step": 18400, "loss": 0.004368708468973637}
+{"method": "flowmo", "step": 18600, "loss": 0.004182927776128054}
+{"method": "flowmo", "step": 18800, "loss": 0.004190036095678806}
+{"method": "flowmo", "step": 19000, "loss": 0.004384973086416721}
+{"method": "flowmo", "step": 19200, "loss": 0.006023879628628492}
+{"method": "flowmo", "step": 19400, "loss": 0.004392072558403015}
+{"method": "flowmo", "step": 19600, "loss": 0.004413294140249491}
+{"method": "flowmo", "step": 19800, "loss": 0.004305647686123848}
+{"method": "flowmo", "step": 20000, "loss": 0.004372047260403633}
diff --git a/experiments/flowmo/result/parameter_count.json b/experiments/flowmo/result/parameter_count.json
new file mode 100644
index 0000000000000000000000000000000000000000..8a396acaa8040fd7c0578e8b63296f0c4440b963
--- /dev/null
+++ b/experiments/flowmo/result/parameter_count.json
@@ -0,0 +1,11 @@
+{
+  "encoder": 340416,
+  "state_history": 75648,
+  "context_history": 75648,
+  "to_z": 30960,
+  "to_c": 17544,
+  "base_delta": 45808,
+  "residual_delta": 46448,
+  "decoder": 31492,
+  "total": 663964
+}
\ No newline at end of file
diff --git a/experiments/leworldmodel/result/paper_training_trace.jsonl b/experiments/leworldmodel/result/paper_training_trace.jsonl
new file mode 100644
index 0000000000000000000000000000000000000000..175f060035d07997813cfa0d5d17b78948eb472d
--- /dev/null
+++ b/experiments/leworldmodel/result/paper_training_trace.jsonl
@@ -0,0 +1,100 @@
+{"method": "leworldmodel", "step": 200, "loss": 2.569669246673584}
+{"method": "leworldmodel", "step": 400, "loss": 2.5372421741485596}
+{"method": "leworldmodel", "step": 600, "loss": 2.037346363067627}
+{"method": "leworldmodel", "step": 800, "loss": 1.8627333641052246}
+{"method": "leworldmodel", "step": 1000, "loss": 1.8265268802642822}
+{"method": "leworldmodel", "step": 1200, "loss": 1.7777857780456543}
+{"method": "leworldmodel", "step": 1400, "loss": 1.753949522972107}
+{"method": "leworldmodel", "step": 1600, "loss": 1.7200227975845337}
+{"method": "leworldmodel", "step": 1800, "loss": 1.6717208623886108}
+{"method": "leworldmodel", "step": 2000, "loss": 1.391864538192749}
+{"method": "leworldmodel", "step": 2200, "loss": 0.7442419528961182}
+{"method": "leworldmodel", "step": 2400, "loss": 0.2299843579530716}
+{"method": "leworldmodel", "step": 2600, "loss": 0.13865336775779724}
+{"method": "leworldmodel", "step": 2800, "loss": 0.09015313535928726}
+{"method": "leworldmodel", "step": 3000, "loss": 0.06993035972118378}
+{"method": "leworldmodel", "step": 3200, "loss": 0.05781794339418411}
+{"method": "leworldmodel", "step": 3400, "loss": 0.05077105388045311}
+{"method": "leworldmodel", "step": 3600, "loss": 0.043729424476623535}
+{"method": "leworldmodel", "step": 3800, "loss": 0.040191106498241425}
+{"method": "leworldmodel", "step": 4000, "loss": 0.03818749263882637}
+{"method": "leworldmodel", "step": 4200, "loss": 0.03584111109375954}
+{"method": "leworldmodel", "step": 4400, "loss": 0.03268589824438095}
+{"method": "leworldmodel", "step": 4600, "loss": 0.030512923374772072}
+{"method": "leworldmodel", "step": 4800, "loss": 0.028514141216874123}
+{"method": "leworldmodel", "step": 5000, "loss": 0.026582585647702217}
+{"method": "leworldmodel", "step": 5200, "loss": 0.02657574787735939}
+{"method": "leworldmodel", "step": 5400, "loss": 0.03360811248421669}
+{"method": "leworldmodel", "step": 5600, "loss": 0.024245228618383408}
+{"method": "leworldmodel", "step": 5800, "loss": 0.025151818990707397}
+{"method": "leworldmodel", "step": 6000, "loss": 0.02470393292605877}
+{"method": "leworldmodel", "step": 6200, "loss": 0.022554941475391388}
+{"method": "leworldmodel", "step": 6400, "loss": 0.021832682192325592}
+{"method": "leworldmodel", "step": 6600, "loss": 0.02138935960829258}
+{"method": "leworldmodel", "step": 6800, "loss": 0.021714694797992706}
+{"method": "leworldmodel", "step": 7000, "loss": 0.022361399605870247}
+{"method": "leworldmodel", "step": 7200, "loss": 0.021948281675577164}
+{"method": "leworldmodel", "step": 7400, "loss": 0.023297373205423355}
+{"method": "leworldmodel", "step": 7600, "loss": 0.01953437551856041}
+{"method": "leworldmodel", "step": 7800, "loss": 0.019761236384510994}
+{"method": "leworldmodel", "step": 8000, "loss": 0.018553584814071655}
+{"method": "leworldmodel", "step": 8200, "loss": 0.01823526993393898}
+{"method": "leworldmodel", "step": 8400, "loss": 0.01854352466762066}
+{"method": "leworldmodel", "step": 8600, "loss": 0.019154751673340797}
+{"method": "leworldmodel", "step": 8800, "loss": 0.018928799778223038}
+{"method": "leworldmodel", "step": 9000, "loss": 0.01909957453608513}
+{"method": "leworldmodel", "step": 9200, "loss": 0.018046768382191658}
+{"method": "leworldmodel", "step": 9400, "loss": 0.016244517639279366}
+{"method": "leworldmodel", "step": 9600, "loss": 0.016833283007144928}
+{"method": "leworldmodel", "step": 9800, "loss": 0.01727704517543316}
+{"method": "leworldmodel", "step": 10000, "loss": 0.017903970554471016}
+{"method": "leworldmodel", "step": 10200, "loss": 0.01650647632777691}
+{"method": "leworldmodel", "step": 10400, "loss": 0.015822188928723335}
+{"method": "leworldmodel", "step": 10600, "loss": 0.021761486306786537}
+{"method": "leworldmodel", "step": 10800, "loss": 0.01572641171514988}
+{"method": "leworldmodel", "step": 11000, "loss": 0.014744052663445473}
+{"method": "leworldmodel", "step": 11200, "loss": 0.014771764166653156}
+{"method": "leworldmodel", "step": 11400, "loss": 0.015161859802901745}
+{"method": "leworldmodel", "step": 11600, "loss": 0.015039228834211826}
+{"method": "leworldmodel", "step": 11800, "loss": 0.01450162474066019}
+{"method": "leworldmodel", "step": 12000, "loss": 0.014639433473348618}
+{"method": "leworldmodel", "step": 12200, "loss": 0.014432272873818874}
+{"method": "leworldmodel", "step": 12400, "loss": 0.05746567249298096}
+{"method": "leworldmodel", "step": 12600, "loss": 0.01567252166569233}
+{"method": "leworldmodel", "step": 12800, "loss": 0.013239766471087933}
+{"method": "leworldmodel", "step": 13000, "loss": 0.01337014976888895}
+{"method": "leworldmodel", "step": 13200, "loss": 0.013945686630904675}
+{"method": "leworldmodel", "step": 13400, "loss": 0.013215066865086555}
+{"method": "leworldmodel", "step": 13600, "loss": 0.013161891140043736}
+{"method": "leworldmodel", "step": 13800, "loss": 0.013161612674593925}
+{"method": "leworldmodel", "step": 14000, "loss": 0.013272494077682495}
+{"method": "leworldmodel", "step": 14200, "loss": 0.012501145713031292}
+{"method": "leworldmodel", "step": 14400, "loss": 0.01319703459739685}
+{"method": "leworldmodel", "step": 14600, "loss": 0.01253820862621069}
+{"method": "leworldmodel", "step": 14800, "loss": 0.013268169946968555}
+{"method": "leworldmodel", "step": 15000, "loss": 0.012286090292036533}
+{"method": "leworldmodel", "step": 15200, "loss": 0.012689301744103432}
+{"method": "leworldmodel", "step": 15400, "loss": 0.018598034977912903}
+{"method": "leworldmodel", "step": 15600, "loss": 0.011223368346691132}
+{"method": "leworldmodel", "step": 15800, "loss": 0.011263682506978512}
+{"method": "leworldmodel", "step": 16000, "loss": 0.011280846782028675}
+{"method": "leworldmodel", "step": 16200, "loss": 0.011259369552135468}
+{"method": "leworldmodel", "step": 16400, "loss": 0.012492901645600796}
+{"method": "leworldmodel", "step": 16600, "loss": 0.011446918360888958}
+{"method": "leworldmodel", "step": 16800, "loss": 0.01105540618300438}
+{"method": "leworldmodel", "step": 17000, "loss": 0.010708491317927837}
+{"method": "leworldmodel", "step": 17200, "loss": 0.01076575368642807}
+{"method": "leworldmodel", "step": 17400, "loss": 0.010710487142205238}
+{"method": "leworldmodel", "step": 17600, "loss": 0.010570250451564789}
+{"method": "leworldmodel", "step": 17800, "loss": 0.010576384142041206}
+{"method": "leworldmodel", "step": 18000, "loss": 0.011023864150047302}
+{"method": "leworldmodel", "step": 18200, "loss": 0.010448881424963474}
+{"method": "leworldmodel", "step": 18400, "loss": 0.010450177825987339}
+{"method": "leworldmodel", "step": 18600, "loss": 0.010127122513949871}
+{"method": "leworldmodel", "step": 18800, "loss": 0.010028455406427383}
+{"method": "leworldmodel", "step": 19000, "loss": 0.010146670043468475}
+{"method": "leworldmodel", "step": 19200, "loss": 0.010228784754872322}
+{"method": "leworldmodel", "step": 19400, "loss": 0.009929011575877666}
+{"method": "leworldmodel", "step": 19600, "loss": 0.009595287963747978}
+{"method": "leworldmodel", "step": 19800, "loss": 0.00958178285509348}
+{"method": "leworldmodel", "step": 20000, "loss": 0.018198927864432335}
diff --git a/experiments/planet/checkpoint/paper.pt b/experiments/planet/checkpoint/paper.pt
new file mode 100644
index 0000000000000000000000000000000000000000..211515c3c9d15a7e5184a0f64d107242e0009b18
--- /dev/null
+++ b/experiments/planet/checkpoint/paper.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ad00b1fb32dbdf8bd8e3346f33809449b3e322b2d8c63215122b4d2b6037281
+size 2667969
diff --git a/experiments/planet/checkpoint/paper_step_002000.pt b/experiments/planet/checkpoint/paper_step_002000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..0ee1184479a5111089297c5e35db9371f4753913
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_002000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a87a5d79f98b826efa88e576a3bb99b4bfa048594a8e2609d817ea8f3d679c10
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_004000.pt b/experiments/planet/checkpoint/paper_step_004000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..0f7619ea9ad192b02c2fdcac22d22ceb4a12ceb8
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_004000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cc6062cf4afdd18c3695e82390d999649e17ae3d1df1df9516b98c932a18a76
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_006000.pt b/experiments/planet/checkpoint/paper_step_006000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..0363ad7047202ab90cb35f7e18dfb6c970027805
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_006000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c431debb323790b30e21ce0cf2880da57ec068e2774768a06bb97378d7b35039
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_008000.pt b/experiments/planet/checkpoint/paper_step_008000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..d5ff622c50d887f9bc20f466eb05d6c8e7eedc02
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_008000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9bcddf47b999152542af212d9f2fa29c219d08b086cd373f69d09c3068590f21
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_010000.pt b/experiments/planet/checkpoint/paper_step_010000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..4d2a5dabaef611029b480de5f9426c5041496087
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_010000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25e3fb22c8d47deeb820a0432f46a4fd06129816388de15f9fcbff5c1ecd13ff
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_012000.pt b/experiments/planet/checkpoint/paper_step_012000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..c9aca920de947ab95e7a3b78edde01f71ea390b4
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_012000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fd9663fe0a8da38a2aed2fc3f724bce338fa648ba670b4008e5d321cb0cd24ef
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_014000.pt b/experiments/planet/checkpoint/paper_step_014000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..92763f0ac7911275a6f007440449cacaae353440
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_014000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c7a2aa2430399471f0f96a2ed971ebfada00642963db63f7fe8a27063972326
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_016000.pt b/experiments/planet/checkpoint/paper_step_016000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..3fc09a953f7aac0731c0d536106804be61749bca
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_016000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1a796fa6a832bea8ac98e0abb81a1bdf33ea6abd3ed0a2bedd97d3a1e8755be
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_018000.pt b/experiments/planet/checkpoint/paper_step_018000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..52bc70f3a25b8ee62603fe7a2da9aa1f1f56538d
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_018000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:593427167eb8fb37c7c8176b4a2010816168be5202c3d2b122b4aca74e54f653
+size 2669553
diff --git a/experiments/planet/checkpoint/paper_step_020000.pt b/experiments/planet/checkpoint/paper_step_020000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..e2274796ce61a8fc8a91962832df55e39c97e3ee
--- /dev/null
+++ b/experiments/planet/checkpoint/paper_step_020000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5801d52adb4e2b64221343aef7ab7d887e4d87d1637a4209df46579cd9d022a6
+size 2669553
diff --git a/experiments/reports/paper_planning/gifs/image_planning_current_estimator_mpc_inferred_triangle_waypoint_zigzag_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_current_estimator_mpc_inferred_triangle_waypoint_zigzag_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..7ce3578acf308221fbc56ccbf462ffc886d0925a
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_current_estimator_mpc_inferred_triangle_waypoint_zigzag_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76ab59bf05174d46c5f2748f6ed8f728cef3566b01e8488c4e0e49bd56336239
+size 177341
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_counterflow_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_counterflow_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..ef2969c49f85ee8677cc2af390c0eeed0e7a6d1a
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_counterflow_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:836e45a83daa8167d754f9acc819ab3bf7342a7e473ccf449ed9a6c82402e620
+size 169178
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_waypoint_zigzag_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_waypoint_zigzag_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..0d0840541e6555142f6085abb84a6da9af459417
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_triangle_waypoint_zigzag_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7acf9539ca1b67a88e2f2d8939c4b2caaaace7dc555f4f4bd1aabfb464effa4d
+size 390692
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_passive_to_active_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_passive_to_active_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..70b05b9ce6ca1fdb2d4862e5d48afb34cbd94469
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_passive_to_active_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c46f812aebd9acd84219a5356108099a907d382d00cd0c61903211d692f5e87e
+size 188300
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_square_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_square_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1d4b7fac65350766da0e37231bc17eb06d5a40f6
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_square_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd18f6d3443710e139f44ab8fd6595aac5c9621b4eacc317134ef76c2ccb79c1
+size 194355
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_zigzag_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_zigzag_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8665061c2551b7da157a000b577adedb238efba6
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_inferred_twin_waypoint_zigzag_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4a28dee8a7a4a5f74596bf75a51adf6b5c3d5cc2936f28b6687c3fb4b04dad9
+size 311193
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_reach_uniform_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_reach_uniform_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..cb809bcc7ffacaea9c503b5517ce8b24bbe5379b
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_reach_uniform_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec97f00a6b919444f36c0eecc204369a0fe016fe07cdfb500e2ebe67a9e4f842
+size 168016
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_station_keeping_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_station_keeping_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..9f684716f5836e0468c5d028c3f7923538ebc4dc
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_station_keeping_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:90feb6ce07570b9aa35d098b8ca5cbb87279d8984f5bb6e5137f121e2d951c53
+size 188495
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2176e3a674e56e6ace769d294321730eda616929
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e2937423176df47b539ac85da10b622277c75bc51a07c75c118fbc5285ea5f55
+size 121532
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..44d3c202adf1b629671d06fe29eea2dccb9f7927
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_triangle_waypoint_square_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d502c8b9aeeeee9ebf18784f7d13d7bc9d7a32b6f126da3f79663df0c62f8746
+size 115829
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_passive_to_active_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_passive_to_active_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..88437f5b3c3525907a325fa9d450f124c64ad4a5
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_passive_to_active_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51579501bd1e2c65d0ae150159ba5cee6a43ecdf0e6edc644ea6df6ff8418a44
+size 136825
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4ff9994b3d7dc94975b09a0c1db4812acf2df079
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:815f1d63abf1a19b0500d0ad376534d7379143e5e9edbe78026ff6339eecea71
+size 133533
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..cec6dcb7033673566b1e86d2e4d0cccce335f45f
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_reach_uniform_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b95a5e1f72a998a233a7888111a38e2f6825976bfd3c8669710c86ce7d166af
+size 262732
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_waypoint_zigzag_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_waypoint_zigzag_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..c637001067697338cdb2aac8247ffdb0dc9e78db
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_shuffled_twin_waypoint_zigzag_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e9eef89a623e14b8145b1543cc948a76a1aee589783b786740635ce9f381ba5
+size 569638
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_passive_to_active_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_passive_to_active_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1d527ebec336f544ab0b0a59d85a3307160c3ea1
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_passive_to_active_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a9174691102568439b0a3ce83159a2275611cbba44f01642c2bf138fa245175
+size 134205
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_reach_uniform_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_reach_uniform_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2253b4c333b48dc967028c0a3f91865fd35ebf06
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_reach_uniform_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:495cf648e12233b1226b7a848e1e16f03fe0695cbd6269b6687cc4c4acf52e3f
+size 135763
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_waypoint_zigzag_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_waypoint_zigzag_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4bcc90d091509cac462c7ddf89ba470894bcfc1e
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_triangle_waypoint_zigzag_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e39c3cc1159886ad0329fbc6ec6479239bffc6b22527201fe622c41f0fc8aaa1
+size 500439
diff --git a/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_twin_waypoint_zigzag_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_twin_waypoint_zigzag_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1b045393f03c0125a3fb658217b0559302f1cf21
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_flowmo_zero_twin_waypoint_zigzag_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:92de50f17a5955ee56ff7f2e2d1b739f189ba9cf8a535f4537f42ae093318b0c
+size 556850
diff --git a/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_reach_uniform_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_reach_uniform_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..b2678f77e3849eb91ecb337a3fe61dd03465d47e
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_reach_uniform_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5367ff8c4cf89a5c8e0ce26935e50a5ca5e13a0ad0315e79095781d1cb2e46b
+size 179732
diff --git a/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_square_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_square_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a81ea9e633befbf394ad397e71e251107b520956
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_square_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e8a6409984e8798e876e60635d9cd26c0cbe98e9fe41acce8bc4639f95476fa8
+size 258209
diff --git a/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_zigzag_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_zigzag_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2e32487185978df70c3f314bf79d3a63bd41825d
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_triangle_waypoint_zigzag_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:80b2b81ff28f64f93931467a07c6a08b22f4b0addef56b0a41abb1fa1876dc82
+size 507061
diff --git a/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_twin_passive_to_active_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_twin_passive_to_active_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8eabdce33142f475bec50f930771d99526c0ecdf
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_leworldmodel_inferred_twin_passive_to_active_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74115a946f1d24944826c6942f9a5f5677810da02b561e3cbfaca3721d7dbb90
+size 194443
diff --git a/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..49b563675c67782dd50c83908b0746a3c7a9510d
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dcf66be535f831e17f4b94118db98f28466166e82b9931c46f6dd0050d977e2f
+size 179119
diff --git a/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..164ff517fe0268ef21c3bc81d9bd88888678cb47
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_square_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3c4f5035d89e7bc2ddee91520189ef37be2174188aa02254cb574bc313931794
+size 169547
diff --git a/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_zigzag_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_zigzag_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..da5d5e37f1ef1be2c2ff03eba5cb25228b3bafb2
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_triangle_waypoint_zigzag_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b204f886163f469254a0e2ccce9916b6fd5a753fea6520abfc57055f9b2f019
+size 269285
diff --git a/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_twin_waypoint_zigzag_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_twin_waypoint_zigzag_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..fb6c4defc6b95d186ba8e423cf45460c2a6d4018
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_pid_los_controller_inferred_twin_waypoint_zigzag_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a614c0e6a4aec7ba261680b06a5718cc83b0114fe8408e238d2ceb5360697adc
+size 304723
diff --git a/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..b067e441eb0187c69881e958726196ab55ee95ea
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d7f64e35752eeb8bc40b07b710e33ff3d53747227b8369b4b830ab22d3f3d8eb
+size 204080
diff --git a/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1c01cf3204322b8fb1895cefbb04ef5778e75af6
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_reach_uniform_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d97ff305120c49270d519766edfa18894b9c905c2319fbce136cf5936c7393e7
+size 204350
diff --git a/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_square_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_square_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4e7bd02df247fea9a81ccd5fc642250b363a70e7
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_square_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9aa4ba34e07d63578d7e5cfe8dbd68420c6a8b15fde24e74a17c86ecdb8e4ca3
+size 253432
diff --git a/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_zigzag_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_zigzag_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..6cdb1025e367b30b231cf8440b791993c1b0648a
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_triangle_waypoint_zigzag_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed49743b2b79c68b745a6c6277c1f9555e091b52042dc41fb2bbd351dc191323
+size 667297
diff --git a/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..dd27b86e2c24b9c8952a58a86feae1ed0b97a579
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7bf423a19b576a21aac2d0b2ab8c55631712d11104d419bbb06fb7a75a1e8ebe
+size 197178
diff --git a/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..bed65d5097cb382cea8bbff819d78ea72f6a33aa
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_passive_to_active_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:753dcc4affbdbc0b96a957986aca4e5482a59e0261fd99f09eeae2af4f494e1f
+size 332091
diff --git a/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_waypoint_square_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_waypoint_square_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..5f29aa94231861435193113b94605fdcaaaabde6
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_planet_inferred_twin_waypoint_square_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7eef15a3774ddeaed3773da0cb2d5faa7e7f835422e82cb365b404ec7ae91266
+size 294610
diff --git a/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_passive_to_active_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_passive_to_active_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..366ffd6292fd2a1fb121055de05ffeee68644282
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_passive_to_active_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d46151561752e9e315b7e983fdd5c87d317b8633b74f47360f4a115d39d59c44
+size 321151
diff --git a/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_square_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_square_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..9eae14cd23769735c1b2507197cfbf82165e4d37
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_square_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:32bed6cd5367ffd3969013cb0079858ebcbca2f0c41921de45d0ac661cb5c5cd
+size 282368
diff --git a/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_zigzag_ep002.gif b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_zigzag_ep002.gif
new file mode 100644
index 0000000000000000000000000000000000000000..7a547a95c56dbb31d2d0d1fa539613d4a6f9d55f
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_triangle_waypoint_zigzag_ep002.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e557ca464356a06667930d77863c2a835ebd438810d9872b7b93c92350f593ca
+size 371960
diff --git a/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..80024e40375e141ea3777d5702e44e3e98d7b24b
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a076a6b4813d5b200a56da44d02d40fa674f957bca0ce82351deda9abda2e5d7
+size 210743
diff --git a/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2c81c6c463c0db84f1c99d415c641962d527c558
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_passive_to_active_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c16176638a5b57dfa370df0ea7d8ed90b7b4203199fa7980972ccc01342ff553
+size 390662
diff --git a/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep000.gif b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep000.gif
new file mode 100644
index 0000000000000000000000000000000000000000..7fc27e416d338da593873296bc4a6e2aea0a442a
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep000.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2292fc98fe7b2c2af918095776b9484165898a4d85b955cc2cb7ec7158be49e6
+size 507583
diff --git a/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep001.gif b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep001.gif
new file mode 100644
index 0000000000000000000000000000000000000000..abc058074ea8e10959dce67904cf3c10330bda23
--- /dev/null
+++ b/experiments/reports/paper_planning/gifs/image_planning_tdmpc2_inferred_twin_waypoint_zigzag_ep001.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8e5f15295b5caec8c318e96b31d513088b0a001c0fac85cbb70197cb2522924
+size 525749
diff --git a/experiments/shared/config/paper_image.json b/experiments/shared/config/paper_image.json
index 721501eb489f8fe3170e1931fdae7f0d9897c318..f17d114bacbfdf66697faac7a9b3f82e05307e39 100644
--- a/experiments/shared/config/paper_image.json
+++ b/experiments/shared/config/paper_image.json
@@ -87,21 +87,24 @@
       "waypoint_zigzag": 900
     },
     "make_gifs": 3,
-    "planner": "cem",
-    "planner_iterations": 30,
-    "planner_lr": 0.08,
-    "cem_horizon": 30,
+    "cem_horizon": 45,
     "cem_population": 512,
     "cem_elites": 64,
     "cem_iterations": 4,
     "cem_action_std": 0.5,
-    "cem_knots": 8,
-    "cem_w_route": 4.0,
+    "cem_knots": 10,
+    "cem_w_goal": 6.0,
+    "cem_w_path": 0.2,
+    "cem_w_route": 6.0,
+    "cem_w_lookahead": 2.0,
+    "cem_w_via": 2.0,
+    "cem_route_horizon_distance": 3.0,
     "cem_w_heading_goal": 0.0,
-    "cem_w_progress": 0.0,
+    "cem_w_progress": 2.5,
     "cem_w_action": 0.08,
     "cem_w_smooth": 0.08,
-    "cem_w_boundary": 250.0
+    "cem_w_boundary": 250.0,
+    "cem_boundary_margin": 0.75
   },
   "probe_eval": {
     "train_source": "data/paper/train.npz",
diff --git a/experiments/tdmpc2/checkpoint/paper_step_010000.pt b/experiments/tdmpc2/checkpoint/paper_step_010000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..25c17952716abaf4896eca213f5c8f87e2deb129
--- /dev/null
+++ b/experiments/tdmpc2/checkpoint/paper_step_010000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e1751aa5b2a5158d844a9879b45081ec34aad173c6b3303455d2bc00711251d7
+size 2682538
diff --git a/experiments/tdmpc2/checkpoint/paper_step_014000.pt b/experiments/tdmpc2/checkpoint/paper_step_014000.pt
new file mode 100644
index 0000000000000000000000000000000000000000..76e27a14221bc012b53f2d7b2cd47c30c1512718
--- /dev/null
+++ b/experiments/tdmpc2/checkpoint/paper_step_014000.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8102bda17305492acbe41828cd7ab75f514b7ee3572d5f533b85a8bcc1b1ae43
+size 2682538
diff --git a/experiments/tdmpc2/result/paper_training.json b/experiments/tdmpc2/result/paper_training.json
new file mode 100644
index 0000000000000000000000000000000000000000..0bed272f1ac26e64438ddba19c1652bba8da4edb
--- /dev/null
+++ b/experiments/tdmpc2/result/paper_training.json
@@ -0,0 +1,43 @@
+{
+  "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
+  }
+}
\ No newline at end of file
diff --git a/experiments/tdmpc2/result/paper_training_trace.jsonl b/experiments/tdmpc2/result/paper_training_trace.jsonl
new file mode 100644
index 0000000000000000000000000000000000000000..a099dc4b1a41a0aa7f7b9dbb0fcdef0593a0b689
--- /dev/null
+++ b/experiments/tdmpc2/result/paper_training_trace.jsonl
@@ -0,0 +1,100 @@
+{"method": "tdmpc2", "step": 200, "loss": 2.5615713596343994}
+{"method": "tdmpc2", "step": 400, "loss": 2.5611941814422607}
+{"method": "tdmpc2", "step": 600, "loss": 2.5563809871673584}
+{"method": "tdmpc2", "step": 800, "loss": 2.1208419799804688}
+{"method": "tdmpc2", "step": 1000, "loss": 1.8078142404556274}
+{"method": "tdmpc2", "step": 1200, "loss": 1.7206906080245972}
+{"method": "tdmpc2", "step": 1400, "loss": 1.6611711978912354}
+{"method": "tdmpc2", "step": 1600, "loss": 1.6124956607818604}
+{"method": "tdmpc2", "step": 1800, "loss": 1.5684468746185303}
+{"method": "tdmpc2", "step": 2000, "loss": 1.473024845123291}
+{"method": "tdmpc2", "step": 2200, "loss": 0.9605032801628113}
+{"method": "tdmpc2", "step": 2400, "loss": 0.20805805921554565}
+{"method": "tdmpc2", "step": 2600, "loss": 0.10994510352611542}
+{"method": "tdmpc2", "step": 2800, "loss": 0.07723093777894974}
+{"method": "tdmpc2", "step": 3000, "loss": 0.06331751495599747}
+{"method": "tdmpc2", "step": 3200, "loss": 0.059253498911857605}
+{"method": "tdmpc2", "step": 3400, "loss": 0.0509064719080925}
+{"method": "tdmpc2", "step": 3600, "loss": 0.05013246461749077}
+{"method": "tdmpc2", "step": 3800, "loss": 0.044939637184143066}
+{"method": "tdmpc2", "step": 4000, "loss": 0.04123403877019882}
+{"method": "tdmpc2", "step": 4200, "loss": 0.038444504141807556}
+{"method": "tdmpc2", "step": 4400, "loss": 0.03545379638671875}
+{"method": "tdmpc2", "step": 4600, "loss": 0.033392250537872314}
+{"method": "tdmpc2", "step": 4800, "loss": 0.03270633518695831}
+{"method": "tdmpc2", "step": 5000, "loss": 0.03083067201077938}
+{"method": "tdmpc2", "step": 5200, "loss": 0.029741723090410233}
+{"method": "tdmpc2", "step": 5400, "loss": 0.028314776718616486}
+{"method": "tdmpc2", "step": 5600, "loss": 0.030176477506756783}
+{"method": "tdmpc2", "step": 5800, "loss": 0.025744978338479996}
+{"method": "tdmpc2", "step": 6000, "loss": 0.024305839091539383}
+{"method": "tdmpc2", "step": 6200, "loss": 0.023685595020651817}
+{"method": "tdmpc2", "step": 6400, "loss": 0.02259833738207817}
+{"method": "tdmpc2", "step": 6600, "loss": 0.02283419854938984}
+{"method": "tdmpc2", "step": 6800, "loss": 0.02146834135055542}
+{"method": "tdmpc2", "step": 7000, "loss": 0.022132400423288345}
+{"method": "tdmpc2", "step": 7200, "loss": 0.02122395671904087}
+{"method": "tdmpc2", "step": 7400, "loss": 0.02004498615860939}
+{"method": "tdmpc2", "step": 7600, "loss": 0.020597226917743683}
+{"method": "tdmpc2", "step": 7800, "loss": 0.019778165966272354}
+{"method": "tdmpc2", "step": 8000, "loss": 0.028772568330168724}
+{"method": "tdmpc2", "step": 8200, "loss": 0.019369708374142647}
+{"method": "tdmpc2", "step": 8400, "loss": 0.018060458824038506}
+{"method": "tdmpc2", "step": 8600, "loss": 0.017832478508353233}
+{"method": "tdmpc2", "step": 8800, "loss": 0.017115240916609764}
+{"method": "tdmpc2", "step": 9000, "loss": 0.018560046330094337}
+{"method": "tdmpc2", "step": 9200, "loss": 0.02106262929737568}
+{"method": "tdmpc2", "step": 9400, "loss": 0.0185258686542511}
+{"method": "tdmpc2", "step": 9600, "loss": 0.016134586185216904}
+{"method": "tdmpc2", "step": 9800, "loss": 0.016202867031097412}
+{"method": "tdmpc2", "step": 10000, "loss": 0.01595011167228222}
+{"method": "tdmpc2", "step": 10200, "loss": 0.018244240432977676}
+{"method": "tdmpc2", "step": 10400, "loss": 0.015602342784404755}
+{"method": "tdmpc2", "step": 10600, "loss": 0.01539525855332613}
+{"method": "tdmpc2", "step": 10800, "loss": 0.016917798668146133}
+{"method": "tdmpc2", "step": 11000, "loss": 0.015368577092885971}
+{"method": "tdmpc2", "step": 11200, "loss": 0.01462937705218792}
+{"method": "tdmpc2", "step": 11400, "loss": 0.015103993006050587}
+{"method": "tdmpc2", "step": 11600, "loss": 0.022605471312999725}
+{"method": "tdmpc2", "step": 11800, "loss": 0.014880099333822727}
+{"method": "tdmpc2", "step": 12000, "loss": 0.014849118888378143}
+{"method": "tdmpc2", "step": 12200, "loss": 0.013959258794784546}
+{"method": "tdmpc2", "step": 12400, "loss": 0.013942420482635498}
+{"method": "tdmpc2", "step": 12600, "loss": 0.014197458513081074}
+{"method": "tdmpc2", "step": 12800, "loss": 0.013932747766375542}
+{"method": "tdmpc2", "step": 13000, "loss": 0.020875083282589912}
+{"method": "tdmpc2", "step": 13200, "loss": 0.013686371967196465}
+{"method": "tdmpc2", "step": 13400, "loss": 0.013075173832476139}
+{"method": "tdmpc2", "step": 13600, "loss": 0.013095398433506489}
+{"method": "tdmpc2", "step": 13800, "loss": 0.013131505809724331}
+{"method": "tdmpc2", "step": 14000, "loss": 0.013600788079202175}
+{"method": "tdmpc2", "step": 14200, "loss": 0.013556594960391521}
+{"method": "tdmpc2", "step": 14400, "loss": 0.013314628042280674}
+{"method": "tdmpc2", "step": 14600, "loss": 0.012985733337700367}
+{"method": "tdmpc2", "step": 14800, "loss": 0.012749229557812214}
+{"method": "tdmpc2", "step": 15000, "loss": 0.01288189459592104}
+{"method": "tdmpc2", "step": 15200, "loss": 0.012339785695075989}
+{"method": "tdmpc2", "step": 15400, "loss": 0.012346968054771423}
+{"method": "tdmpc2", "step": 15600, "loss": 0.011947316117584705}
+{"method": "tdmpc2", "step": 15800, "loss": 0.019620070233941078}
+{"method": "tdmpc2", "step": 16000, "loss": 0.016492800787091255}
+{"method": "tdmpc2", "step": 16200, "loss": 0.01130885723978281}
+{"method": "tdmpc2", "step": 16400, "loss": 0.011348055675625801}
+{"method": "tdmpc2", "step": 16600, "loss": 0.01124806609004736}
+{"method": "tdmpc2", "step": 16800, "loss": 0.011437931098043919}
+{"method": "tdmpc2", "step": 17000, "loss": 0.011486472561955452}
+{"method": "tdmpc2", "step": 17200, "loss": 0.011247489601373672}
+{"method": "tdmpc2", "step": 17400, "loss": 0.010947943665087223}
+{"method": "tdmpc2", "step": 17600, "loss": 0.010786485858261585}
+{"method": "tdmpc2", "step": 17800, "loss": 0.011596443131566048}
+{"method": "tdmpc2", "step": 18000, "loss": 0.011309945024549961}
+{"method": "tdmpc2", "step": 18200, "loss": 0.01111777313053608}
+{"method": "tdmpc2", "step": 18400, "loss": 0.010638328269124031}
+{"method": "tdmpc2", "step": 18600, "loss": 0.010363967157900333}
+{"method": "tdmpc2", "step": 18800, "loss": 0.01042043324559927}
+{"method": "tdmpc2", "step": 19000, "loss": 0.010521506890654564}
+{"method": "tdmpc2", "step": 19200, "loss": 0.010812968946993351}
+{"method": "tdmpc2", "step": 19400, "loss": 0.010421177372336388}
+{"method": "tdmpc2", "step": 19600, "loss": 0.010399093851447105}
+{"method": "tdmpc2", "step": 19800, "loss": 0.0101461885496974}
+{"method": "tdmpc2", "step": 20000, "loss": 0.010143937543034554}
diff --git a/experiments/tdmpc2/result/parameter_count.json b/experiments/tdmpc2/result/parameter_count.json
new file mode 100644
index 0000000000000000000000000000000000000000..d680d6036a7e7dd18c37b049d6ec93ee14c42b46
--- /dev/null
+++ b/experiments/tdmpc2/result/parameter_count.json
@@ -0,0 +1,8 @@
+{
+  "encoder": 471584,
+  "history": 87936,
+  "to_z": 30960,
+  "transition": 45808,
+  "decoder": 31492,
+  "total": 667780
+}
\ No newline at end of file