Hugging Face's logo

cccat6
/
FlowMo-WM

Robotics
PyTorch
world-model
visual-world-model
model-based-control
surface-vehicle
hidden-drift
Model card Files
xet
 Community
cccat6 commited on
Commit
38965f7
·
verified ·
1 Parent(s): 9e39d2a

Sync FlowMo training metadata

Browse files
Files changed (3) hide show
  1. experiments/flowmo/result/paper_training.json +43 -0
  2. experiments/flowmo/result/paper_training_trace.jsonl +100 -0
  3. experiments/flowmo/result/parameter_count.json +11 -0
experiments/flowmo/result/paper_training.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "flowmo",
3
+ "steps": 20000,
4
+ "batch_size": 256,
5
+ "train_samples": 5120000,
6
+ "final_train_loss": 0.003838971257209778,
7
+ "total_parameters": 663964,
8
+ "target_mode": "absolute_normalized",
9
+ "position_scale": 5.0,
10
+ "heading_weight": 2.0,
11
+ "current_pose_weight": 1.0,
12
+ "motion_weight": 0.5,
13
+ "precision": "bf16",
14
+ "checkpoint_name": "paper.pt",
15
+ "final_checkpoint": "paper.pt",
16
+ "intermediate_checkpoints": [
17
+ "paper_step_002000.pt",
18
+ "paper_step_004000.pt",
19
+ "paper_step_006000.pt",
20
+ "paper_step_008000.pt",
21
+ "paper_step_010000.pt",
22
+ "paper_step_012000.pt",
23
+ "paper_step_014000.pt",
24
+ "paper_step_016000.pt",
25
+ "paper_step_018000.pt",
26
+ "paper_step_020000.pt"
27
+ ],
28
+ "checkpoint_interval": 2000,
29
+ "prediction": {
30
+ "pos1": 0.07454978911361347,
31
+ "heading1": 0.04916680386910836,
32
+ "pos3": 0.08039923206282158,
33
+ "heading3": 0.05139941743497426,
34
+ "pos6": 0.09065966509903471,
35
+ "heading6": 0.05502280278597027,
36
+ "pos8": 0.09813052571068208,
37
+ "heading8": 0.05761814404589435,
38
+ "pos10": 0.10625893513982494,
39
+ "heading10": 0.060202051497374974,
40
+ "pos20": 0.15069053896392384,
41
+ "heading20": 0.07164489043255647
42
+ }
43
+ }
experiments/flowmo/result/paper_training_trace.jsonl ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"method": "flowmo", "step": 200, "loss": 2.5674960613250732}
2
+ {"method": "flowmo", "step": 400, "loss": 2.5546326637268066}
3
+ {"method": "flowmo", "step": 600, "loss": 2.1885855197906494}
4
+ {"method": "flowmo", "step": 800, "loss": 1.9521373510360718}
5
+ {"method": "flowmo", "step": 1000, "loss": 1.8136850595474243}
6
+ {"method": "flowmo", "step": 1200, "loss": 1.7078666687011719}
7
+ {"method": "flowmo", "step": 1400, "loss": 1.583600401878357}
8
+ {"method": "flowmo", "step": 1600, "loss": 1.4113436937332153}
9
+ {"method": "flowmo", "step": 1800, "loss": 1.1391433477401733}
10
+ {"method": "flowmo", "step": 2000, "loss": 0.843991756439209}
11
+ {"method": "flowmo", "step": 2200, "loss": 0.4143493175506592}
12
+ {"method": "flowmo", "step": 2400, "loss": 0.17534112930297852}
13
+ {"method": "flowmo", "step": 2600, "loss": 0.10404947400093079}
14
+ {"method": "flowmo", "step": 2800, "loss": 0.08960390090942383}
15
+ {"method": "flowmo", "step": 3000, "loss": 0.06150925159454346}
16
+ {"method": "flowmo", "step": 3200, "loss": 0.050643716007471085}
17
+ {"method": "flowmo", "step": 3400, "loss": 0.04722023382782936}
18
+ {"method": "flowmo", "step": 3600, "loss": 0.03910943493247032}
19
+ {"method": "flowmo", "step": 3800, "loss": 0.034517355263233185}
20
+ {"method": "flowmo", "step": 4000, "loss": 0.03372931107878685}
21
+ {"method": "flowmo", "step": 4200, "loss": 0.028907420113682747}
22
+ {"method": "flowmo", "step": 4400, "loss": 0.02713683620095253}
23
+ {"method": "flowmo", "step": 4600, "loss": 0.026512157171964645}
24
+ {"method": "flowmo", "step": 4800, "loss": 0.02357216738164425}
25
+ {"method": "flowmo", "step": 5000, "loss": 0.022205041721463203}
26
+ {"method": "flowmo", "step": 5200, "loss": 0.021671824157238007}
27
+ {"method": "flowmo", "step": 5400, "loss": 0.022753609344363213}
28
+ {"method": "flowmo", "step": 5600, "loss": 0.017359044402837753}
29
+ {"method": "flowmo", "step": 5800, "loss": 0.015896229073405266}
30
+ {"method": "flowmo", "step": 6000, "loss": 0.015997033566236496}
31
+ {"method": "flowmo", "step": 6200, "loss": 0.017708348110318184}
32
+ {"method": "flowmo", "step": 6400, "loss": 0.013344906270503998}
33
+ {"method": "flowmo", "step": 6600, "loss": 0.019510427489876747}
34
+ {"method": "flowmo", "step": 6800, "loss": 0.014104553498327732}
35
+ {"method": "flowmo", "step": 7000, "loss": 0.012482959777116776}
36
+ {"method": "flowmo", "step": 7200, "loss": 0.020980888977646828}
37
+ {"method": "flowmo", "step": 7400, "loss": 0.010919320397078991}
38
+ {"method": "flowmo", "step": 7600, "loss": 0.012034476734697819}
39
+ {"method": "flowmo", "step": 7800, "loss": 0.010465476661920547}
40
+ {"method": "flowmo", "step": 8000, "loss": 0.010064244270324707}
41
+ {"method": "flowmo", "step": 8200, "loss": 0.017180416733026505}
42
+ {"method": "flowmo", "step": 8400, "loss": 0.012871088460087776}
43
+ {"method": "flowmo", "step": 8600, "loss": 0.009764967486262321}
44
+ {"method": "flowmo", "step": 8800, "loss": 0.009405333548784256}
45
+ {"method": "flowmo", "step": 9000, "loss": 0.008828608319163322}
46
+ {"method": "flowmo", "step": 9200, "loss": 0.008842001669108868}
47
+ {"method": "flowmo", "step": 9400, "loss": 0.012081452645361423}
48
+ {"method": "flowmo", "step": 9600, "loss": 0.018835758790373802}
49
+ {"method": "flowmo", "step": 9800, "loss": 0.008033945225179195}
50
+ {"method": "flowmo", "step": 10000, "loss": 0.007763190194964409}
51
+ {"method": "flowmo", "step": 10200, "loss": 0.007489423733204603}
52
+ {"method": "flowmo", "step": 10400, "loss": 0.0076056718826293945}
53
+ {"method": "flowmo", "step": 10600, "loss": 0.00768063310533762}
54
+ {"method": "flowmo", "step": 10800, "loss": 0.009992506355047226}
55
+ {"method": "flowmo", "step": 11000, "loss": 0.00842261966317892}
56
+ {"method": "flowmo", "step": 11200, "loss": 0.026521500200033188}
57
+ {"method": "flowmo", "step": 11400, "loss": 0.018860813230276108}
58
+ {"method": "flowmo", "step": 11600, "loss": 0.007891213521361351}
59
+ {"method": "flowmo", "step": 11800, "loss": 0.006570508703589439}
60
+ {"method": "flowmo", "step": 12000, "loss": 0.006496565416455269}
61
+ {"method": "flowmo", "step": 12200, "loss": 0.006572996266186237}
62
+ {"method": "flowmo", "step": 12400, "loss": 0.006286568008363247}
63
+ {"method": "flowmo", "step": 12600, "loss": 0.006442747078835964}
64
+ {"method": "flowmo", "step": 12800, "loss": 0.0060729458928108215}
65
+ {"method": "flowmo", "step": 13000, "loss": 0.006215339060872793}
66
+ {"method": "flowmo", "step": 13200, "loss": 0.006229731719940901}
67
+ {"method": "flowmo", "step": 13400, "loss": 0.015923455357551575}
68
+ {"method": "flowmo", "step": 13600, "loss": 0.00830706488341093}
69
+ {"method": "flowmo", "step": 13800, "loss": 0.005948069971054792}
70
+ {"method": "flowmo", "step": 14000, "loss": 0.005477099679410458}
71
+ {"method": "flowmo", "step": 14200, "loss": 0.005825021769851446}
72
+ {"method": "flowmo", "step": 14400, "loss": 0.005389831960201263}
73
+ {"method": "flowmo", "step": 14600, "loss": 0.00538972020149231}
74
+ {"method": "flowmo", "step": 14800, "loss": 0.005436031147837639}
75
+ {"method": "flowmo", "step": 15000, "loss": 0.005460439249873161}
76
+ {"method": "flowmo", "step": 15200, "loss": 0.0054905107244849205}
77
+ {"method": "flowmo", "step": 15400, "loss": 0.04125838354229927}
78
+ {"method": "flowmo", "step": 15600, "loss": 0.007642415817826986}
79
+ {"method": "flowmo", "step": 15800, "loss": 0.004961329512298107}
80
+ {"method": "flowmo", "step": 16000, "loss": 0.0046252962201833725}
81
+ {"method": "flowmo", "step": 16200, "loss": 0.004764193668961525}
82
+ {"method": "flowmo", "step": 16400, "loss": 0.0046986970119178295}
83
+ {"method": "flowmo", "step": 16600, "loss": 0.004872818943113089}
84
+ {"method": "flowmo", "step": 16800, "loss": 0.004637495614588261}
85
+ {"method": "flowmo", "step": 17000, "loss": 0.004717609845101833}
86
+ {"method": "flowmo", "step": 17200, "loss": 0.00468458840623498}
87
+ {"method": "flowmo", "step": 17400, "loss": 0.004698699805885553}
88
+ {"method": "flowmo", "step": 17600, "loss": 0.0048277308233082294}
89
+ {"method": "flowmo", "step": 17800, "loss": 0.00458098016679287}
90
+ {"method": "flowmo", "step": 18000, "loss": 0.0046404642052948475}
91
+ {"method": "flowmo", "step": 18200, "loss": 0.004587466362863779}
92
+ {"method": "flowmo", "step": 18400, "loss": 0.004678289871662855}
93
+ {"method": "flowmo", "step": 18600, "loss": 0.004505776334553957}
94
+ {"method": "flowmo", "step": 18800, "loss": 0.004538910463452339}
95
+ {"method": "flowmo", "step": 19000, "loss": 0.03361465781927109}
96
+ {"method": "flowmo", "step": 19200, "loss": 0.007749940734356642}
97
+ {"method": "flowmo", "step": 19400, "loss": 0.004111349582672119}
98
+ {"method": "flowmo", "step": 19600, "loss": 0.003949436359107494}
99
+ {"method": "flowmo", "step": 19800, "loss": 0.0038776814471930265}
100
+ {"method": "flowmo", "step": 20000, "loss": 0.003838971257209778}
experiments/flowmo/result/parameter_count.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "encoder": 340416,
3
+ "state_history": 75648,
4
+ "context_history": 75648,
5
+ "to_z": 30960,
6
+ "to_c": 17544,
7
+ "base_delta": 45808,
8
+ "residual_delta": 46448,
9
+ "decoder": 31492,
10
+ "total": 663964
11
+ }