Pthahnix commited on
Commit
fcbdc6e
·
verified ·
1 Parent(s): 01780e7

Upload checkpoints/exp4_B_lvis_wide/training_history.json with huggingface_hub

Browse files
checkpoints/exp4_B_lvis_wide/training_history.json ADDED
@@ -0,0 +1,2602 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "epoch": 0,
4
+ "loss": 0.42678292487371905,
5
+ "recon_loss": 0.384222788377829,
6
+ "commit_loss": 0.021280068076319165,
7
+ "embed_loss": 0.021280068076319165,
8
+ "codebook_utilization": 0.95458984375,
9
+ "lr": 2.08e-05,
10
+ "phase": "full_vq",
11
+ "val_recon_loss": 0.3549533709046546,
12
+ "val_utilization": 0.805908203125,
13
+ "time_sec": 145.14698576927185
14
+ },
15
+ {
16
+ "epoch": 1,
17
+ "loss": 0.285142692245119,
18
+ "recon_loss": 0.25313153241061903,
19
+ "commit_loss": 0.01600558005480303,
20
+ "embed_loss": 0.01600558005480303,
21
+ "codebook_utilization": 0.896484375,
22
+ "lr": 4.06e-05,
23
+ "phase": "full_vq",
24
+ "val_recon_loss": 0.23466047937615533,
25
+ "val_utilization": 0.8154296875,
26
+ "time_sec": 146.8876917362213
27
+ },
28
+ {
29
+ "epoch": 2,
30
+ "loss": 0.26130804716329264,
31
+ "recon_loss": 0.2269260234181797,
32
+ "commit_loss": 0.017191012015157357,
33
+ "embed_loss": 0.017191012015157357,
34
+ "codebook_utilization": 0.950439453125,
35
+ "lr": 6.04e-05,
36
+ "phase": "full_vq",
37
+ "val_recon_loss": 0.22493889082348748,
38
+ "val_utilization": 0.9267578125,
39
+ "time_sec": 144.71639704704285
40
+ },
41
+ {
42
+ "epoch": 3,
43
+ "loss": 0.2609999136226933,
44
+ "recon_loss": 0.22009544456150473,
45
+ "commit_loss": 0.02045223454068993,
46
+ "embed_loss": 0.02045223454068993,
47
+ "codebook_utilization": 0.95947265625,
48
+ "lr": 8.02e-05,
49
+ "phase": "full_vq",
50
+ "val_recon_loss": 0.22049254254343803,
51
+ "val_utilization": 0.940673828125,
52
+ "time_sec": 138.95289158821106
53
+ },
54
+ {
55
+ "epoch": 4,
56
+ "loss": 0.2613859515647255,
57
+ "recon_loss": 0.2171086357618735,
58
+ "commit_loss": 0.022138657906473815,
59
+ "embed_loss": 0.022138657906473815,
60
+ "codebook_utilization": 0.96142578125,
61
+ "lr": 0.0001,
62
+ "phase": "full_vq",
63
+ "val_recon_loss": 0.21841969533582753,
64
+ "val_utilization": 0.949951171875,
65
+ "time_sec": 142.81331729888916
66
+ },
67
+ {
68
+ "epoch": 5,
69
+ "loss": 0.26106645808956486,
70
+ "recon_loss": 0.2160396533526056,
71
+ "commit_loss": 0.022513402542629377,
72
+ "embed_loss": 0.022513402542629377,
73
+ "codebook_utilization": 0.96142578125,
74
+ "lr": 9.999351124856874e-05,
75
+ "phase": "full_vq",
76
+ "val_recon_loss": 0.21808490952414072,
77
+ "val_utilization": 0.95068359375,
78
+ "time_sec": 140.55375170707703
79
+ },
80
+ {
81
+ "epoch": 6,
82
+ "loss": 0.26045460122710645,
83
+ "recon_loss": 0.215332179312622,
84
+ "commit_loss": 0.022561211255063534,
85
+ "embed_loss": 0.022561211255063534,
86
+ "codebook_utilization": 0.961669921875,
87
+ "lr": 9.997404667843075e-05,
88
+ "phase": "full_vq",
89
+ "val_recon_loss": 0.21696466718162044,
90
+ "val_utilization": 0.94921875,
91
+ "time_sec": 139.82509326934814
92
+ },
93
+ {
94
+ "epoch": 7,
95
+ "loss": 0.25995056803633526,
96
+ "recon_loss": 0.21480332214011733,
97
+ "commit_loss": 0.022573623190404277,
98
+ "embed_loss": 0.022573623190404277,
99
+ "codebook_utilization": 0.9619140625,
100
+ "lr": 9.994161134161632e-05,
101
+ "phase": "full_vq",
102
+ "val_recon_loss": 0.21736847550681468,
103
+ "val_utilization": 0.951171875,
104
+ "time_sec": 145.66957640647888
105
+ },
106
+ {
107
+ "epoch": 8,
108
+ "loss": 0.2597095783204244,
109
+ "recon_loss": 0.21441424920710767,
110
+ "commit_loss": 0.02264766449608454,
111
+ "embed_loss": 0.02264766449608454,
112
+ "codebook_utilization": 0.9619140625,
113
+ "lr": 9.9896213656719e-05,
114
+ "phase": "full_vq",
115
+ "val_recon_loss": 0.21668462078557926,
116
+ "val_utilization": 0.949462890625,
117
+ "time_sec": 140.21882152557373
118
+ },
119
+ {
120
+ "epoch": 9,
121
+ "loss": 0.2595992099624985,
122
+ "recon_loss": 0.21429500318962708,
123
+ "commit_loss": 0.022652103101233965,
124
+ "embed_loss": 0.022652103101233965,
125
+ "codebook_utilization": 0.9619140625,
126
+ "lr": 9.98378654067105e-05,
127
+ "phase": "full_vq",
128
+ "val_recon_loss": 0.21648672884434797,
129
+ "val_utilization": 0.94775390625,
130
+ "time_sec": 143.0625777244568
131
+ },
132
+ {
133
+ "epoch": 10,
134
+ "loss": 0.25944911959129296,
135
+ "recon_loss": 0.21404535749417333,
136
+ "commit_loss": 0.022701881381715864,
137
+ "embed_loss": 0.022701881381715864,
138
+ "codebook_utilization": 0.9619140625,
139
+ "lr": 9.976658173588243e-05,
140
+ "phase": "full_vq",
141
+ "val_recon_loss": 0.21621907293126824,
142
+ "val_utilization": 0.94775390625,
143
+ "time_sec": 137.69888043403625
144
+ },
145
+ {
146
+ "epoch": 11,
147
+ "loss": 0.25926318191254366,
148
+ "recon_loss": 0.21395548521907026,
149
+ "commit_loss": 0.02265384854612554,
150
+ "embed_loss": 0.02265384854612554,
151
+ "codebook_utilization": 0.961669921875,
152
+ "lr": 9.968238114591565e-05,
153
+ "phase": "full_vq",
154
+ "val_recon_loss": 0.21603316051906415,
155
+ "val_utilization": 0.95068359375,
156
+ "time_sec": 142.64535641670227
157
+ },
158
+ {
159
+ "epoch": 12,
160
+ "loss": 0.2590584269141763,
161
+ "recon_loss": 0.21374999202767686,
162
+ "commit_loss": 0.02265421743063019,
163
+ "embed_loss": 0.02265421743063019,
164
+ "codebook_utilization": 0.96142578125,
165
+ "lr": 9.95852854910781e-05,
166
+ "phase": "full_vq",
167
+ "val_recon_loss": 0.21634705029846577,
168
+ "val_utilization": 0.94921875,
169
+ "time_sec": 142.26861476898193
170
+ },
171
+ {
172
+ "epoch": 13,
173
+ "loss": 0.25916050956016634,
174
+ "recon_loss": 0.21358976949278902,
175
+ "commit_loss": 0.02278537000592566,
176
+ "embed_loss": 0.02278537000592566,
177
+ "codebook_utilization": 0.961669921875,
178
+ "lr": 9.947531997255256e-05,
179
+ "phase": "full_vq",
180
+ "val_recon_loss": 0.2170534807811962,
181
+ "val_utilization": 0.9462890625,
182
+ "time_sec": 141.56940078735352
183
+ },
184
+ {
185
+ "epoch": 14,
186
+ "loss": 0.2593067276082065,
187
+ "recon_loss": 0.2136510015745473,
188
+ "commit_loss": 0.02282786319350324,
189
+ "embed_loss": 0.02282786319350324,
190
+ "codebook_utilization": 0.9619140625,
191
+ "lr": 9.935251313189564e-05,
192
+ "phase": "full_vq",
193
+ "val_recon_loss": 0.21626966145266308,
194
+ "val_utilization": 0.95166015625,
195
+ "time_sec": 148.4954264163971
196
+ },
197
+ {
198
+ "epoch": 15,
199
+ "loss": 0.2590380136559649,
200
+ "recon_loss": 0.21337022684857773,
201
+ "commit_loss": 0.02283389314120701,
202
+ "embed_loss": 0.02283389314120701,
203
+ "codebook_utilization": 0.961669921875,
204
+ "lr": 9.921689684362989e-05,
205
+ "phase": "full_vq",
206
+ "val_recon_loss": 0.2162867784500122,
207
+ "val_utilization": 0.95166015625,
208
+ "time_sec": 142.23319745063782
209
+ },
210
+ {
211
+ "epoch": 16,
212
+ "loss": 0.25910502442014893,
213
+ "recon_loss": 0.21332048614658314,
214
+ "commit_loss": 0.022892268808674683,
215
+ "embed_loss": 0.022892268808674683,
216
+ "codebook_utilization": 0.961669921875,
217
+ "lr": 9.906850630697068e-05,
218
+ "phase": "full_vq",
219
+ "val_recon_loss": 0.21574814081861732,
220
+ "val_utilization": 0.949951171875,
221
+ "time_sec": 140.62019729614258
222
+ },
223
+ {
224
+ "epoch": 17,
225
+ "loss": 0.25915054993093173,
226
+ "recon_loss": 0.21333036392038754,
227
+ "commit_loss": 0.02291009295227001,
228
+ "embed_loss": 0.02291009295227001,
229
+ "codebook_utilization": 0.9619140625,
230
+ "lr": 9.890738003669029e-05,
231
+ "phase": "full_vq",
232
+ "val_recon_loss": 0.21606976957468504,
233
+ "val_utilization": 0.94970703125,
234
+ "time_sec": 146.04079604148865
235
+ },
236
+ {
237
+ "epoch": 18,
238
+ "loss": 0.2587732972733697,
239
+ "recon_loss": 0.2131749229621758,
240
+ "commit_loss": 0.02279918707987965,
241
+ "embed_loss": 0.02279918707987965,
242
+ "codebook_utilization": 0.9619140625,
243
+ "lr": 9.87335598531214e-05,
244
+ "phase": "full_vq",
245
+ "val_recon_loss": 0.21590039226111402,
246
+ "val_utilization": 0.948486328125,
247
+ "time_sec": 140.32899260520935
248
+ },
249
+ {
250
+ "epoch": 19,
251
+ "loss": 0.258949479456678,
252
+ "recon_loss": 0.21307361774085984,
253
+ "commit_loss": 0.022937930726665792,
254
+ "embed_loss": 0.022937930726665792,
255
+ "codebook_utilization": 0.961669921875,
256
+ "lr": 9.85470908713026e-05,
257
+ "phase": "full_vq",
258
+ "val_recon_loss": 0.21569685746779602,
259
+ "val_utilization": 0.94482421875,
260
+ "time_sec": 138.57794523239136
261
+ },
262
+ {
263
+ "epoch": 20,
264
+ "loss": 0.25896169140248443,
265
+ "recon_loss": 0.21299062925706388,
266
+ "commit_loss": 0.02298553139324676,
267
+ "embed_loss": 0.02298553139324676,
268
+ "codebook_utilization": 0.96142578125,
269
+ "lr": 9.834802148926882e-05,
270
+ "phase": "full_vq",
271
+ "val_recon_loss": 0.21607097656874175,
272
+ "val_utilization": 0.94091796875,
273
+ "time_sec": 146.97264623641968
274
+ },
275
+ {
276
+ "epoch": 21,
277
+ "loss": 0.25897987857259064,
278
+ "recon_loss": 0.21294317467183602,
279
+ "commit_loss": 0.023018351912518664,
280
+ "embed_loss": 0.023018351912518664,
281
+ "codebook_utilization": 0.96142578125,
282
+ "lr": 9.813640337548953e-05,
283
+ "phase": "full_vq",
284
+ "val_recon_loss": 0.21521069670326254,
285
+ "val_utilization": 0.947509765625,
286
+ "time_sec": 140.86066317558289
287
+ },
288
+ {
289
+ "epoch": 22,
290
+ "loss": 0.2590546062806757,
291
+ "recon_loss": 0.2129275978298045,
292
+ "commit_loss": 0.02306350421533998,
293
+ "embed_loss": 0.02306350421533998,
294
+ "codebook_utilization": 0.961669921875,
295
+ "lr": 9.79122914554583e-05,
296
+ "phase": "full_vq",
297
+ "val_recon_loss": 0.21561545133590698,
298
+ "val_utilization": 0.947509765625,
299
+ "time_sec": 143.34402227401733
300
+ },
301
+ {
302
+ "epoch": 23,
303
+ "loss": 0.25874617903778546,
304
+ "recon_loss": 0.21280971262351608,
305
+ "commit_loss": 0.022968232952219846,
306
+ "embed_loss": 0.022968232952219846,
307
+ "codebook_utilization": 0.961669921875,
308
+ "lr": 9.76757438974368e-05,
309
+ "phase": "full_vq",
310
+ "val_recon_loss": 0.21556306839658973,
311
+ "val_utilization": 0.947998046875,
312
+ "time_sec": 141.0192894935608
313
+ },
314
+ {
315
+ "epoch": 24,
316
+ "loss": 0.2590210713183654,
317
+ "recon_loss": 0.2128203742832026,
318
+ "commit_loss": 0.023100348398957634,
319
+ "embed_loss": 0.023100348398957634,
320
+ "codebook_utilization": 0.961181640625,
321
+ "lr": 9.742682209735726e-05,
322
+ "phase": "full_vq",
323
+ "val_recon_loss": 0.21521995836094523,
324
+ "val_utilization": 0.943603515625,
325
+ "time_sec": 141.3150978088379
326
+ },
327
+ {
328
+ "epoch": 25,
329
+ "loss": 0.25883271066638514,
330
+ "recon_loss": 0.21271629746609588,
331
+ "commit_loss": 0.023058206658194542,
332
+ "embed_loss": 0.023058206658194542,
333
+ "codebook_utilization": 0.961669921875,
334
+ "lr": 9.716559066288714e-05,
335
+ "phase": "full_vq",
336
+ "val_recon_loss": 0.21569039446584296,
337
+ "val_utilization": 0.947509765625,
338
+ "time_sec": 143.95515966415405
339
+ },
340
+ {
341
+ "epoch": 26,
342
+ "loss": 0.2586478032878421,
343
+ "recon_loss": 0.21258624570398796,
344
+ "commit_loss": 0.023030778605157766,
345
+ "embed_loss": 0.023030778605157766,
346
+ "codebook_utilization": 0.961669921875,
347
+ "lr": 9.689211739666022e-05,
348
+ "phase": "full_vq",
349
+ "val_recon_loss": 0.21528827248329527,
350
+ "val_utilization": 0.947998046875,
351
+ "time_sec": 138.48145580291748
352
+ },
353
+ {
354
+ "epoch": 27,
355
+ "loss": 0.25868336026503785,
356
+ "recon_loss": 0.21251595268646875,
357
+ "commit_loss": 0.02308370378676065,
358
+ "embed_loss": 0.02308370378676065,
359
+ "codebook_utilization": 0.96142578125,
360
+ "lr": 9.660647327867839e-05,
361
+ "phase": "full_vq",
362
+ "val_recon_loss": 0.21476852492000278,
363
+ "val_utilization": 0.94677734375,
364
+ "time_sec": 139.43713068962097
365
+ },
366
+ {
367
+ "epoch": 28,
368
+ "loss": 0.25879625592451433,
369
+ "recon_loss": 0.21250995525176608,
370
+ "commit_loss": 0.023143150583717235,
371
+ "embed_loss": 0.023143150583717235,
372
+ "codebook_utilization": 0.9609375,
373
+ "lr": 9.630873244788881e-05,
374
+ "phase": "full_vq",
375
+ "val_recon_loss": 0.21509383367688467,
376
+ "val_utilization": 0.949462890625,
377
+ "time_sec": 139.26381301879883
378
+ },
379
+ {
380
+ "epoch": 29,
381
+ "loss": 0.2587965873441076,
382
+ "recon_loss": 0.21244799850513618,
383
+ "commit_loss": 0.023174294543157264,
384
+ "embed_loss": 0.023174294543157264,
385
+ "codebook_utilization": 0.96142578125,
386
+ "lr": 9.59989721829412e-05,
387
+ "phase": "full_vq",
388
+ "val_recon_loss": 0.21496142999509746,
389
+ "val_utilization": 0.95068359375,
390
+ "time_sec": 137.41713452339172
391
+ },
392
+ {
393
+ "epoch": 30,
394
+ "loss": 0.25870538700806095,
395
+ "recon_loss": 0.21239799463005893,
396
+ "commit_loss": 0.023153696292481287,
397
+ "embed_loss": 0.023153696292481287,
398
+ "codebook_utilization": 0.96142578125,
399
+ "lr": 9.567727288213003e-05,
400
+ "phase": "full_vq",
401
+ "val_recon_loss": 0.21520997182037052,
402
+ "val_utilization": 0.946044921875,
403
+ "time_sec": 142.72812342643738
404
+ },
405
+ {
406
+ "epoch": 31,
407
+ "loss": 0.25869924569033026,
408
+ "recon_loss": 0.21226186040778794,
409
+ "commit_loss": 0.023218692674081984,
410
+ "embed_loss": 0.023218692674081984,
411
+ "codebook_utilization": 0.96142578125,
412
+ "lr": 9.534371804252726e-05,
413
+ "phase": "full_vq",
414
+ "val_recon_loss": 0.2152991117386336,
415
+ "val_utilization": 0.946044921875,
416
+ "time_sec": 139.53481817245483
417
+ },
418
+ {
419
+ "epoch": 32,
420
+ "loss": 0.25865321043061046,
421
+ "recon_loss": 0.21233445054631891,
422
+ "commit_loss": 0.02315938008096079,
423
+ "embed_loss": 0.02315938008096079,
424
+ "codebook_utilization": 0.9609375,
425
+ "lr": 9.499839423831059e-05,
426
+ "phase": "full_vq",
427
+ "val_recon_loss": 0.2147088528014301,
428
+ "val_utilization": 0.947021484375,
429
+ "time_sec": 139.7909595966339
430
+ },
431
+ {
432
+ "epoch": 33,
433
+ "loss": 0.2586699183273122,
434
+ "recon_loss": 0.21219236694458055,
435
+ "commit_loss": 0.023238775464213962,
436
+ "embed_loss": 0.023238775464213962,
437
+ "codebook_utilization": 0.96142578125,
438
+ "lr": 9.464139109829319e-05,
439
+ "phase": "full_vq",
440
+ "val_recon_loss": 0.2149084819836563,
441
+ "val_utilization": 0.945556640625,
442
+ "time_sec": 142.13117337226868
443
+ },
444
+ {
445
+ "epoch": 34,
446
+ "loss": 0.25861272116905953,
447
+ "recon_loss": 0.21216259565618303,
448
+ "commit_loss": 0.023225062832155526,
449
+ "embed_loss": 0.023225062832155526,
450
+ "codebook_utilization": 0.96142578125,
451
+ "lr": 9.427280128266047e-05,
452
+ "phase": "full_vq",
453
+ "val_recon_loss": 0.21517228494199475,
454
+ "val_utilization": 0.950439453125,
455
+ "time_sec": 140.5292046070099
456
+ },
457
+ {
458
+ "epoch": 35,
459
+ "loss": 0.25849526807625445,
460
+ "recon_loss": 0.2120135441462845,
461
+ "commit_loss": 0.0232408620558457,
462
+ "embed_loss": 0.0232408620558457,
463
+ "codebook_utilization": 0.960693359375,
464
+ "lr": 9.389272045892021e-05,
465
+ "phase": "full_vq",
466
+ "val_recon_loss": 0.21490926747576575,
467
+ "val_utilization": 0.9453125,
468
+ "time_sec": 143.545307636261
469
+ },
470
+ {
471
+ "epoch": 36,
472
+ "loss": 0.2586581244418615,
473
+ "recon_loss": 0.21205935893746897,
474
+ "commit_loss": 0.023299382835485265,
475
+ "embed_loss": 0.023299382835485265,
476
+ "codebook_utilization": 0.9609375,
477
+ "lr": 9.350124727707194e-05,
478
+ "phase": "full_vq",
479
+ "val_recon_loss": 0.21510211127192785,
480
+ "val_utilization": 0.947265625,
481
+ "time_sec": 141.1629512310028
482
+ },
483
+ {
484
+ "epoch": 37,
485
+ "loss": 0.258615106566327,
486
+ "recon_loss": 0.2119387833648904,
487
+ "commit_loss": 0.023338161504809773,
488
+ "embed_loss": 0.023338161504809773,
489
+ "codebook_utilization": 0.96142578125,
490
+ "lr": 9.309848334400244e-05,
491
+ "phase": "full_vq",
492
+ "val_recon_loss": 0.2153269603346171,
493
+ "val_utilization": 0.939453125,
494
+ "time_sec": 147.63585448265076
495
+ },
496
+ {
497
+ "epoch": 38,
498
+ "loss": 0.2586722240498072,
499
+ "recon_loss": 0.21199430436461275,
500
+ "commit_loss": 0.023338959776975598,
501
+ "embed_loss": 0.023338959776975598,
502
+ "codebook_utilization": 0.960693359375,
503
+ "lr": 9.26845331971136e-05,
504
+ "phase": "full_vq",
505
+ "val_recon_loss": 0.21449511355898354,
506
+ "val_utilization": 0.949951171875,
507
+ "time_sec": 142.93736219406128
508
+ },
509
+ {
510
+ "epoch": 39,
511
+ "loss": 0.25863758840050477,
512
+ "recon_loss": 0.2118038962365489,
513
+ "commit_loss": 0.02341684616274304,
514
+ "embed_loss": 0.02341684616274304,
515
+ "codebook_utilization": 0.9609375,
516
+ "lr": 9.225950427718973e-05,
517
+ "phase": "full_vq",
518
+ "val_recon_loss": 0.21475398925583014,
519
+ "val_utilization": 0.946044921875,
520
+ "time_sec": 142.22387528419495
521
+ },
522
+ {
523
+ "epoch": 40,
524
+ "loss": 0.25843264499734736,
525
+ "recon_loss": 0.21170020844474394,
526
+ "commit_loss": 0.02336621848578618,
527
+ "embed_loss": 0.02336621848578618,
528
+ "codebook_utilization": 0.9609375,
529
+ "lr": 9.182350690051132e-05,
530
+ "phase": "full_vq",
531
+ "val_recon_loss": 0.21437887010279666,
532
+ "val_utilization": 0.948974609375,
533
+ "time_sec": 144.71989679336548
534
+ },
535
+ {
536
+ "epoch": 41,
537
+ "loss": 0.25859518706556256,
538
+ "recon_loss": 0.21173921994723602,
539
+ "commit_loss": 0.023427983501113366,
540
+ "embed_loss": 0.023427983501113366,
541
+ "codebook_utilization": 0.9609375,
542
+ "lr": 9.137665423022247e-05,
543
+ "phase": "full_vq",
544
+ "val_recon_loss": 0.21430128829532794,
545
+ "val_utilization": 0.946533203125,
546
+ "time_sec": 143.93717217445374
547
+ },
548
+ {
549
+ "epoch": 42,
550
+ "loss": 0.2586230036971692,
551
+ "recon_loss": 0.2117257153963655,
552
+ "commit_loss": 0.023448643834913165,
553
+ "embed_loss": 0.023448643834913165,
554
+ "codebook_utilization": 0.9609375,
555
+ "lr": 9.091906224695934e-05,
556
+ "phase": "full_vq",
557
+ "val_recon_loss": 0.21448240738906219,
558
+ "val_utilization": 0.94873046875,
559
+ "time_sec": 144.35693287849426
560
+ },
561
+ {
562
+ "epoch": 43,
563
+ "loss": 0.2585734843319348,
564
+ "recon_loss": 0.21155122911865473,
565
+ "commit_loss": 0.02351112759149656,
566
+ "embed_loss": 0.02351112759149656,
567
+ "codebook_utilization": 0.960693359375,
568
+ "lr": 9.045084971874735e-05,
569
+ "phase": "full_vq",
570
+ "val_recon_loss": 0.2144373216655817,
571
+ "val_utilization": 0.945068359375,
572
+ "time_sec": 144.8519103527069
573
+ },
574
+ {
575
+ "epoch": 44,
576
+ "loss": 0.2586294540221775,
577
+ "recon_loss": 0.21157427350195443,
578
+ "commit_loss": 0.023527590189442078,
579
+ "embed_loss": 0.023527590189442078,
580
+ "codebook_utilization": 0.96044921875,
581
+ "lr": 8.997213817017505e-05,
582
+ "phase": "full_vq",
583
+ "val_recon_loss": 0.2142146984847744,
584
+ "val_utilization": 0.94970703125,
585
+ "time_sec": 142.53119111061096
586
+ },
587
+ {
588
+ "epoch": 45,
589
+ "loss": 0.25842603897741495,
590
+ "recon_loss": 0.21151572439647948,
591
+ "commit_loss": 0.023455157020409418,
592
+ "embed_loss": 0.023455157020409418,
593
+ "codebook_utilization": 0.9609375,
594
+ "lr": 8.948305185085224e-05,
595
+ "phase": "full_vq",
596
+ "val_recon_loss": 0.21423935990655021,
597
+ "val_utilization": 0.94482421875,
598
+ "time_sec": 140.84726548194885
599
+ },
600
+ {
601
+ "epoch": 46,
602
+ "loss": 0.25863368595195657,
603
+ "recon_loss": 0.21144511827285373,
604
+ "commit_loss": 0.023594283821884048,
605
+ "embed_loss": 0.023594283821884048,
606
+ "codebook_utilization": 0.960693359375,
607
+ "lr": 8.89837177031611e-05,
608
+ "phase": "full_vq",
609
+ "val_recon_loss": 0.21399379236979432,
610
+ "val_utilization": 0.947998046875,
611
+ "time_sec": 146.10818243026733
612
+ },
613
+ {
614
+ "epoch": 47,
615
+ "loss": 0.2583680785081897,
616
+ "recon_loss": 0.2113478476721743,
617
+ "commit_loss": 0.023510115370053424,
618
+ "embed_loss": 0.023510115370053424,
619
+ "codebook_utilization": 0.960693359375,
620
+ "lr": 8.847426532930828e-05,
621
+ "phase": "full_vq",
622
+ "val_recon_loss": 0.21420084986459004,
623
+ "val_utilization": 0.94873046875,
624
+ "time_sec": 143.35535621643066
625
+ },
626
+ {
627
+ "epoch": 48,
628
+ "loss": 0.25840087016908136,
629
+ "recon_loss": 0.2113658986723197,
630
+ "commit_loss": 0.023517485937674033,
631
+ "embed_loss": 0.023517485937674033,
632
+ "codebook_utilization": 0.96044921875,
633
+ "lr": 8.795482695768655e-05,
634
+ "phase": "full_vq",
635
+ "val_recon_loss": 0.2138335554954711,
636
+ "val_utilization": 0.949462890625,
637
+ "time_sec": 142.02895545959473
638
+ },
639
+ {
640
+ "epoch": 49,
641
+ "loss": 0.258504908037218,
642
+ "recon_loss": 0.21131342253710841,
643
+ "commit_loss": 0.023595742828295966,
644
+ "embed_loss": 0.023595742828295966,
645
+ "codebook_utilization": 0.960693359375,
646
+ "lr": 8.742553740855503e-05,
647
+ "phase": "full_vq",
648
+ "val_recon_loss": 0.21434635292278248,
649
+ "val_utilization": 0.947509765625,
650
+ "time_sec": 144.69221925735474
651
+ },
652
+ {
653
+ "epoch": 50,
654
+ "loss": 0.25870345726730376,
655
+ "recon_loss": 0.2114958694068397,
656
+ "commit_loss": 0.023603794086714425,
657
+ "embed_loss": 0.023603794086714425,
658
+ "codebook_utilization": 0.960693359375,
659
+ "lr": 8.688653405904649e-05,
660
+ "phase": "full_vq",
661
+ "val_recon_loss": 0.2140408636311467,
662
+ "val_utilization": 0.94970703125,
663
+ "time_sec": 134.34454679489136
664
+ },
665
+ {
666
+ "epoch": 51,
667
+ "loss": 0.2586232081338319,
668
+ "recon_loss": 0.2113049010718418,
669
+ "commit_loss": 0.023659153657190523,
670
+ "embed_loss": 0.023659153657190523,
671
+ "codebook_utilization": 0.960693359375,
672
+ "lr": 8.633795680751113e-05,
673
+ "phase": "full_vq",
674
+ "val_recon_loss": 0.2140832001741013,
675
+ "val_utilization": 0.944580078125,
676
+ "time_sec": 145.97022366523743
677
+ },
678
+ {
679
+ "epoch": 52,
680
+ "loss": 0.25842792470119186,
681
+ "recon_loss": 0.2111025127938123,
682
+ "commit_loss": 0.02366270613288734,
683
+ "embed_loss": 0.02366270613288734,
684
+ "codebook_utilization": 0.960693359375,
685
+ "lr": 8.577994803720604e-05,
686
+ "phase": "full_vq",
687
+ "val_recon_loss": 0.21429527098878046,
688
+ "val_utilization": 0.947998046875,
689
+ "time_sec": 144.35595321655273
690
+ },
691
+ {
692
+ "epoch": 53,
693
+ "loss": 0.25846940246700917,
694
+ "recon_loss": 0.2110294200540558,
695
+ "commit_loss": 0.02371999157496746,
696
+ "embed_loss": 0.02371999157496746,
697
+ "codebook_utilization": 0.96044921875,
698
+ "lr": 8.521265257933945e-05,
699
+ "phase": "full_vq",
700
+ "val_recon_loss": 0.2137241196933757,
701
+ "val_utilization": 0.948486328125,
702
+ "time_sec": 140.27570390701294
703
+ },
704
+ {
705
+ "epoch": 54,
706
+ "loss": 0.2582463869234411,
707
+ "recon_loss": 0.2109071809828766,
708
+ "commit_loss": 0.02366960296271051,
709
+ "embed_loss": 0.02366960296271051,
710
+ "codebook_utilization": 0.960693359375,
711
+ "lr": 8.463621767547996e-05,
712
+ "phase": "full_vq",
713
+ "val_recon_loss": 0.21379048812590287,
714
+ "val_utilization": 0.94970703125,
715
+ "time_sec": 143.96546149253845
716
+ },
717
+ {
718
+ "epoch": 55,
719
+ "loss": 0.25834117385591593,
720
+ "recon_loss": 0.21093789486058037,
721
+ "commit_loss": 0.023701639538050344,
722
+ "embed_loss": 0.023701639538050344,
723
+ "codebook_utilization": 0.960205078125,
724
+ "lr": 8.405079293933984e-05,
725
+ "phase": "full_vq",
726
+ "val_recon_loss": 0.2139232178752342,
727
+ "val_utilization": 0.9482421875,
728
+ "time_sec": 143.47913599014282
729
+ },
730
+ {
731
+ "epoch": 56,
732
+ "loss": 0.25846959803970204,
733
+ "recon_loss": 0.21106102224691772,
734
+ "commit_loss": 0.023704288209356914,
735
+ "embed_loss": 0.023704288209356914,
736
+ "codebook_utilization": 0.960693359375,
737
+ "lr": 8.345653031794289e-05,
738
+ "phase": "full_vq",
739
+ "val_recon_loss": 0.213775457960836,
740
+ "val_utilization": 0.947509765625,
741
+ "time_sec": 142.30744862556458
742
+ },
743
+ {
744
+ "epoch": 57,
745
+ "loss": 0.2583790557373184,
746
+ "recon_loss": 0.21100037981097292,
747
+ "commit_loss": 0.023689338091892116,
748
+ "embed_loss": 0.023689338091892116,
749
+ "codebook_utilization": 0.96044921875,
750
+ "lr": 8.285358405218651e-05,
751
+ "phase": "full_vq",
752
+ "val_recon_loss": 0.21395089569386472,
753
+ "val_utilization": 0.948974609375,
754
+ "time_sec": 135.97141814231873
755
+ },
756
+ {
757
+ "epoch": 58,
758
+ "loss": 0.25882966535766594,
759
+ "recon_loss": 0.2110577279393912,
760
+ "commit_loss": 0.023885968484509443,
761
+ "embed_loss": 0.023885968484509443,
762
+ "codebook_utilization": 0.960693359375,
763
+ "lr": 8.22421106368085e-05,
764
+ "phase": "full_vq",
765
+ "val_recon_loss": 0.2136873584449961,
766
+ "val_utilization": 0.945556640625,
767
+ "time_sec": 142.39403820037842
768
+ },
769
+ {
770
+ "epoch": 59,
771
+ "loss": 0.25817954350454697,
772
+ "recon_loss": 0.21077425856658114,
773
+ "commit_loss": 0.023702642648180488,
774
+ "embed_loss": 0.023702642648180488,
775
+ "codebook_utilization": 0.960693359375,
776
+ "lr": 8.162226877976883e-05,
777
+ "phase": "full_vq",
778
+ "val_recon_loss": 0.21365559184819125,
779
+ "val_utilization": 0.9462890625,
780
+ "time_sec": 145.27197909355164
781
+ },
782
+ {
783
+ "epoch": 60,
784
+ "loss": 0.2584583037065943,
785
+ "recon_loss": 0.21083871343836874,
786
+ "commit_loss": 0.02380979511392149,
787
+ "embed_loss": 0.02380979511392149,
788
+ "codebook_utilization": 0.96044921875,
789
+ "lr": 8.099421936105698e-05,
790
+ "phase": "full_vq",
791
+ "val_recon_loss": 0.21390296979231782,
792
+ "val_utilization": 0.94970703125,
793
+ "time_sec": 130.20773100852966
794
+ },
795
+ {
796
+ "epoch": 61,
797
+ "loss": 0.2585621105865411,
798
+ "recon_loss": 0.21075743192498922,
799
+ "commit_loss": 0.023902339255058667,
800
+ "embed_loss": 0.023902339255058667,
801
+ "codebook_utilization": 0.96044921875,
802
+ "lr": 8.035812539093553e-05,
803
+ "phase": "full_vq",
804
+ "val_recon_loss": 0.2139858975169364,
805
+ "val_utilization": 0.94384765625,
806
+ "time_sec": 129.52642679214478
807
+ },
808
+ {
809
+ "epoch": 62,
810
+ "loss": 0.25831895339779737,
811
+ "recon_loss": 0.21073411310187523,
812
+ "commit_loss": 0.023792420347349914,
813
+ "embed_loss": 0.023792420347349914,
814
+ "codebook_utilization": 0.96044921875,
815
+ "lr": 7.971415196763084e-05,
816
+ "phase": "full_vq",
817
+ "val_recon_loss": 0.21382531172104097,
818
+ "val_utilization": 0.94775390625,
819
+ "time_sec": 145.70008659362793
820
+ },
821
+ {
822
+ "epoch": 63,
823
+ "loss": 0.25849271517216676,
824
+ "recon_loss": 0.210619033109851,
825
+ "commit_loss": 0.023936840708097467,
826
+ "embed_loss": 0.023936840708097467,
827
+ "codebook_utilization": 0.96044921875,
828
+ "lr": 7.90624662344818e-05,
829
+ "phase": "full_vq",
830
+ "val_recon_loss": 0.21374560917696256,
831
+ "val_utilization": 0.947509765625,
832
+ "time_sec": 144.0580291748047
833
+ },
834
+ {
835
+ "epoch": 64,
836
+ "loss": 0.25830113996819754,
837
+ "recon_loss": 0.21055302326756764,
838
+ "commit_loss": 0.023874058105495727,
839
+ "embed_loss": 0.023874058105495727,
840
+ "codebook_utilization": 0.960693359375,
841
+ "lr": 7.840323733655777e-05,
842
+ "phase": "full_vq",
843
+ "val_recon_loss": 0.21365421976935997,
844
+ "val_utilization": 0.949951171875,
845
+ "time_sec": 140.8605101108551
846
+ },
847
+ {
848
+ "epoch": 65,
849
+ "loss": 0.2586238082791085,
850
+ "recon_loss": 0.2107525818595072,
851
+ "commit_loss": 0.023935613315804863,
852
+ "embed_loss": 0.023935613315804863,
853
+ "codebook_utilization": 0.96044921875,
854
+ "lr": 7.773663637675692e-05,
855
+ "phase": "full_vq",
856
+ "val_recon_loss": 0.21348036926114158,
857
+ "val_utilization": 0.94873046875,
858
+ "time_sec": 145.87633538246155
859
+ },
860
+ {
861
+ "epoch": 66,
862
+ "loss": 0.2584222252493305,
863
+ "recon_loss": 0.2104955799776687,
864
+ "commit_loss": 0.023963322749406827,
865
+ "embed_loss": 0.023963322749406827,
866
+ "codebook_utilization": 0.960693359375,
867
+ "lr": 7.706283637139655e-05,
868
+ "phase": "full_vq",
869
+ "val_recon_loss": 0.21327060904730571,
870
+ "val_utilization": 0.947998046875,
871
+ "time_sec": 145.10224032402039
872
+ },
873
+ {
874
+ "epoch": 67,
875
+ "loss": 0.2585794375679357,
876
+ "recon_loss": 0.2106150211438254,
877
+ "commit_loss": 0.023982208176720435,
878
+ "embed_loss": 0.023982208176720435,
879
+ "codebook_utilization": 0.96044921875,
880
+ "lr": 7.638201220530662e-05,
881
+ "phase": "full_vq",
882
+ "val_recon_loss": 0.2136021356066961,
883
+ "val_utilization": 0.945068359375,
884
+ "time_sec": 141.8358600139618
885
+ },
886
+ {
887
+ "epoch": 68,
888
+ "loss": 0.25852942178000604,
889
+ "recon_loss": 0.21052481868079684,
890
+ "commit_loss": 0.024002301706086974,
891
+ "embed_loss": 0.024002301706086974,
892
+ "codebook_utilization": 0.960693359375,
893
+ "lr": 7.569434058643842e-05,
894
+ "phase": "full_vq",
895
+ "val_recon_loss": 0.21368011497379688,
896
+ "val_utilization": 0.94873046875,
897
+ "time_sec": 146.27960991859436
898
+ },
899
+ {
900
+ "epoch": 69,
901
+ "loss": 0.2581202392857572,
902
+ "recon_loss": 0.2104681822544514,
903
+ "commit_loss": 0.023826028376837893,
904
+ "embed_loss": 0.023826028376837893,
905
+ "codebook_utilization": 0.96044921875,
906
+ "lr": 7.499999999999998e-05,
907
+ "phase": "full_vq",
908
+ "val_recon_loss": 0.213332983550061,
909
+ "val_utilization": 0.948486328125,
910
+ "time_sec": 146.26021432876587
911
+ },
912
+ {
913
+ "epoch": 70,
914
+ "loss": 0.25831239628113384,
915
+ "recon_loss": 0.21035886406494672,
916
+ "commit_loss": 0.02397676621157383,
917
+ "embed_loss": 0.02397676621157383,
918
+ "codebook_utilization": 0.96044921875,
919
+ "lr": 7.429917066213028e-05,
920
+ "phase": "full_vq",
921
+ "val_recon_loss": 0.21374644950199662,
922
+ "val_utilization": 0.946533203125,
923
+ "time_sec": 138.32283806800842
924
+ },
925
+ {
926
+ "epoch": 71,
927
+ "loss": 0.25843052186455506,
928
+ "recon_loss": 0.21031855412211198,
929
+ "commit_loss": 0.0240559837525978,
930
+ "embed_loss": 0.0240559837525978,
931
+ "codebook_utilization": 0.96044921875,
932
+ "lr": 7.359203447312409e-05,
933
+ "phase": "full_vq",
934
+ "val_recon_loss": 0.2133876672119237,
935
+ "val_utilization": 0.947509765625,
936
+ "time_sec": 143.6744418144226
937
+ },
938
+ {
939
+ "epoch": 72,
940
+ "loss": 0.25820053522260533,
941
+ "recon_loss": 0.21022797547543276,
942
+ "commit_loss": 0.02398627979029727,
943
+ "embed_loss": 0.02398627979029727,
944
+ "codebook_utilization": 0.96044921875,
945
+ "lr": 7.287877497021975e-05,
946
+ "phase": "full_vq",
947
+ "val_recon_loss": 0.21349885030074067,
948
+ "val_utilization": 0.948974609375,
949
+ "time_sec": 144.0695285797119
950
+ },
951
+ {
952
+ "epoch": 73,
953
+ "loss": 0.2581723588430462,
954
+ "recon_loss": 0.21018956908527106,
955
+ "commit_loss": 0.02399139513127849,
956
+ "embed_loss": 0.02399139513127849,
957
+ "codebook_utilization": 0.960205078125,
958
+ "lr": 7.215957727996207e-05,
959
+ "phase": "full_vq",
960
+ "val_recon_loss": 0.21298488915971156,
961
+ "val_utilization": 0.949462890625,
962
+ "time_sec": 144.38856387138367
963
+ },
964
+ {
965
+ "epoch": 74,
966
+ "loss": 0.25840012991654515,
967
+ "recon_loss": 0.2101793377016618,
968
+ "commit_loss": 0.024110395943387576,
969
+ "embed_loss": 0.024110395943387576,
970
+ "codebook_utilization": 0.960205078125,
971
+ "lr": 7.143462807015271e-05,
972
+ "phase": "full_vq",
973
+ "val_recon_loss": 0.21328262114123012,
974
+ "val_utilization": 0.94873046875,
975
+ "time_sec": 136.60556316375732
976
+ },
977
+ {
978
+ "epoch": 75,
979
+ "loss": 0.2583153761499296,
980
+ "recon_loss": 0.21019017280440344,
981
+ "commit_loss": 0.02406260159452191,
982
+ "embed_loss": 0.02406260159452191,
983
+ "codebook_utilization": 0.96044921875,
984
+ "lr": 7.07041155014006e-05,
985
+ "phase": "full_vq",
986
+ "val_recon_loss": 0.2133090424906002,
987
+ "val_utilization": 0.947265625,
988
+ "time_sec": 134.76925683021545
989
+ },
990
+ {
991
+ "epoch": 76,
992
+ "loss": 0.25833810027941134,
993
+ "recon_loss": 0.2101723048583602,
994
+ "commit_loss": 0.02408289757928228,
995
+ "embed_loss": 0.02408289757928228,
996
+ "codebook_utilization": 0.960205078125,
997
+ "lr": 6.996822917828478e-05,
998
+ "phase": "full_vq",
999
+ "val_recon_loss": 0.21322410369522116,
1000
+ "val_utilization": 0.9443359375,
1001
+ "time_sec": 139.8359179496765
1002
+ },
1003
+ {
1004
+ "epoch": 77,
1005
+ "loss": 0.25834140328937755,
1006
+ "recon_loss": 0.21001468221265773,
1007
+ "commit_loss": 0.024163360707461834,
1008
+ "embed_loss": 0.024163360707461834,
1009
+ "codebook_utilization": 0.96044921875,
1010
+ "lr": 6.922716010014254e-05,
1011
+ "phase": "full_vq",
1012
+ "val_recon_loss": 0.21269190805346777,
1013
+ "val_utilization": 0.946533203125,
1014
+ "time_sec": 143.7204508781433
1015
+ },
1016
+ {
1017
+ "epoch": 78,
1018
+ "loss": 0.25825348664024655,
1019
+ "recon_loss": 0.2099146181490363,
1020
+ "commit_loss": 0.024169434162316122,
1021
+ "embed_loss": 0.024169434162316122,
1022
+ "codebook_utilization": 0.96044921875,
1023
+ "lr": 6.848110061149553e-05,
1024
+ "phase": "full_vq",
1025
+ "val_recon_loss": 0.21288458576028266,
1026
+ "val_utilization": 0.947509765625,
1027
+ "time_sec": 146.03635334968567
1028
+ },
1029
+ {
1030
+ "epoch": 79,
1031
+ "loss": 0.2582908792668565,
1032
+ "recon_loss": 0.20990051547201669,
1033
+ "commit_loss": 0.024195182003424097,
1034
+ "embed_loss": 0.024195182003424097,
1035
+ "codebook_utilization": 0.96044921875,
1036
+ "lr": 6.773024435212678e-05,
1037
+ "phase": "full_vq",
1038
+ "val_recon_loss": 0.21320246377687777,
1039
+ "val_utilization": 0.94775390625,
1040
+ "time_sec": 143.1005551815033
1041
+ },
1042
+ {
1043
+ "epoch": 80,
1044
+ "loss": 0.2582801596380185,
1045
+ "recon_loss": 0.20992169444396244,
1046
+ "commit_loss": 0.024179232761082126,
1047
+ "embed_loss": 0.024179232761082126,
1048
+ "codebook_utilization": 0.960205078125,
1049
+ "lr": 6.697478620682138e-05,
1050
+ "phase": "full_vq",
1051
+ "val_recon_loss": 0.2132864941037103,
1052
+ "val_utilization": 0.950927734375,
1053
+ "time_sec": 142.97854685783386
1054
+ },
1055
+ {
1056
+ "epoch": 81,
1057
+ "loss": 0.2582095956737756,
1058
+ "recon_loss": 0.20976681262254715,
1059
+ "commit_loss": 0.024221391974870107,
1060
+ "embed_loss": 0.024221391974870107,
1061
+ "codebook_utilization": 0.96044921875,
1062
+ "lr": 6.621492225478415e-05,
1063
+ "phase": "full_vq",
1064
+ "val_recon_loss": 0.212753148058827,
1065
+ "val_utilization": 0.947998046875,
1066
+ "time_sec": 142.62985634803772
1067
+ },
1068
+ {
1069
+ "epoch": 82,
1070
+ "loss": 0.2581898486670763,
1071
+ "recon_loss": 0.20979451751563608,
1072
+ "commit_loss": 0.024197665706963398,
1073
+ "embed_loss": 0.024197665706963398,
1074
+ "codebook_utilization": 0.96044921875,
1075
+ "lr": 6.545084971874738e-05,
1076
+ "phase": "full_vq",
1077
+ "val_recon_loss": 0.21299321865767576,
1078
+ "val_utilization": 0.948974609375,
1079
+ "time_sec": 144.02931022644043
1080
+ },
1081
+ {
1082
+ "epoch": 83,
1083
+ "loss": 0.258332510829618,
1084
+ "recon_loss": 0.2098296712730635,
1085
+ "commit_loss": 0.024251419457740737,
1086
+ "embed_loss": 0.024251419457740737,
1087
+ "codebook_utilization": 0.96044921875,
1088
+ "lr": 6.468276691378155e-05,
1089
+ "phase": "full_vq",
1090
+ "val_recon_loss": 0.21286581674318636,
1091
+ "val_utilization": 0.94970703125,
1092
+ "time_sec": 140.63075399398804
1093
+ },
1094
+ {
1095
+ "epoch": 84,
1096
+ "loss": 0.25823936881977044,
1097
+ "recon_loss": 0.20973075193360569,
1098
+ "commit_loss": 0.02425430840017592,
1099
+ "embed_loss": 0.02425430840017592,
1100
+ "codebook_utilization": 0.96044921875,
1101
+ "lr": 6.391087319582264e-05,
1102
+ "phase": "full_vq",
1103
+ "val_recon_loss": 0.21346334443333442,
1104
+ "val_utilization": 0.947509765625,
1105
+ "time_sec": 141.92623496055603
1106
+ },
1107
+ {
1108
+ "epoch": 85,
1109
+ "loss": 0.2583226603544178,
1110
+ "recon_loss": 0.20963993222329028,
1111
+ "commit_loss": 0.024341363813172832,
1112
+ "embed_loss": 0.024341363813172832,
1113
+ "codebook_utilization": 0.960205078125,
1114
+ "lr": 6.313536890992936e-05,
1115
+ "phase": "full_vq",
1116
+ "val_recon_loss": 0.2126564085483551,
1117
+ "val_utilization": 0.94921875,
1118
+ "time_sec": 140.10952138900757
1119
+ },
1120
+ {
1121
+ "epoch": 86,
1122
+ "loss": 0.25829046621393703,
1123
+ "recon_loss": 0.20954461064603594,
1124
+ "commit_loss": 0.02437292785462002,
1125
+ "embed_loss": 0.02437292785462002,
1126
+ "codebook_utilization": 0.96044921875,
1127
+ "lr": 6.23564553382835e-05,
1128
+ "phase": "full_vq",
1129
+ "val_recon_loss": 0.2129648475164778,
1130
+ "val_utilization": 0.947265625,
1131
+ "time_sec": 141.23237371444702
1132
+ },
1133
+ {
1134
+ "epoch": 87,
1135
+ "loss": 0.2582238245664573,
1136
+ "recon_loss": 0.20957176863824126,
1137
+ "commit_loss": 0.024326027835388654,
1138
+ "embed_loss": 0.024326027835388654,
1139
+ "codebook_utilization": 0.96044921875,
1140
+ "lr": 6.157433464794716e-05,
1141
+ "phase": "full_vq",
1142
+ "val_recon_loss": 0.21289081444566169,
1143
+ "val_utilization": 0.949462890625,
1144
+ "time_sec": 138.7257740497589
1145
+ },
1146
+ {
1147
+ "epoch": 88,
1148
+ "loss": 0.258292862716406,
1149
+ "recon_loss": 0.20950252370944192,
1150
+ "commit_loss": 0.024395169538816784,
1151
+ "embed_loss": 0.024395169538816784,
1152
+ "codebook_utilization": 0.960205078125,
1153
+ "lr": 6.0789209838390326e-05,
1154
+ "phase": "full_vq",
1155
+ "val_recon_loss": 0.21318585467472506,
1156
+ "val_utilization": 0.948974609375,
1157
+ "time_sec": 144.73704552650452
1158
+ },
1159
+ {
1160
+ "epoch": 89,
1161
+ "loss": 0.2580897301879679,
1162
+ "recon_loss": 0.20946017473047665,
1163
+ "commit_loss": 0.02431477771864997,
1164
+ "embed_loss": 0.02431477771864997,
1165
+ "codebook_utilization": 0.960205078125,
1166
+ "lr": 6.000128468880224e-05,
1167
+ "phase": "full_vq",
1168
+ "val_recon_loss": 0.21328189876976977,
1169
+ "val_utilization": 0.947265625,
1170
+ "time_sec": 143.4985523223877
1171
+ },
1172
+ {
1173
+ "epoch": 90,
1174
+ "loss": 0.2582075046047286,
1175
+ "recon_loss": 0.2094390815310685,
1176
+ "commit_loss": 0.024384211309678185,
1177
+ "embed_loss": 0.024384211309678185,
1178
+ "codebook_utilization": 0.960205078125,
1179
+ "lr": 5.9210763705200585e-05,
1180
+ "phase": "full_vq",
1181
+ "val_recon_loss": 0.21289943451626916,
1182
+ "val_utilization": 0.947998046875,
1183
+ "time_sec": 137.68448996543884
1184
+ },
1185
+ {
1186
+ "epoch": 91,
1187
+ "loss": 0.25821639738754854,
1188
+ "recon_loss": 0.20935028652107812,
1189
+ "commit_loss": 0.024433055519048116,
1190
+ "embed_loss": 0.024433055519048116,
1191
+ "codebook_utilization": 0.9599609375,
1192
+ "lr": 5.841785206735194e-05,
1193
+ "phase": "full_vq",
1194
+ "val_recon_loss": 0.21300630656520972,
1195
+ "val_utilization": 0.9482421875,
1196
+ "time_sec": 141.46481037139893
1197
+ },
1198
+ {
1199
+ "epoch": 92,
1200
+ "loss": 0.25829135313589724,
1201
+ "recon_loss": 0.2094115660119509,
1202
+ "commit_loss": 0.0244398935468297,
1203
+ "embed_loss": 0.0244398935468297,
1204
+ "codebook_utilization": 0.960205078125,
1205
+ "lr": 5.7622755575517285e-05,
1206
+ "phase": "full_vq",
1207
+ "val_recon_loss": 0.21314790183573626,
1208
+ "val_utilization": 0.94970703125,
1209
+ "time_sec": 143.38036179542542
1210
+ },
1211
+ {
1212
+ "epoch": 93,
1213
+ "loss": 0.25834420910938,
1214
+ "recon_loss": 0.20933896995737625,
1215
+ "commit_loss": 0.02450261960628879,
1216
+ "embed_loss": 0.02450261960628879,
1217
+ "codebook_utilization": 0.9599609375,
1218
+ "lr": 5.6825680597036586e-05,
1219
+ "phase": "full_vq",
1220
+ "val_recon_loss": 0.21262556013096584,
1221
+ "val_utilization": 0.947998046875,
1222
+ "time_sec": 142.0844371318817
1223
+ },
1224
+ {
1225
+ "epoch": 94,
1226
+ "loss": 0.2582503265845097,
1227
+ "recon_loss": 0.20928301839889873,
1228
+ "commit_loss": 0.024483653757125544,
1229
+ "embed_loss": 0.024483653757125544,
1230
+ "codebook_utilization": 0.960205078125,
1231
+ "lr": 5.602683401276615e-05,
1232
+ "phase": "full_vq",
1233
+ "val_recon_loss": 0.21283054862464412,
1234
+ "val_utilization": 0.94873046875,
1235
+ "time_sec": 141.46809554100037
1236
+ },
1237
+ {
1238
+ "epoch": 95,
1239
+ "loss": 0.25828546122079943,
1240
+ "recon_loss": 0.2092043063868352,
1241
+ "commit_loss": 0.02454057753308194,
1242
+ "embed_loss": 0.02454057753308194,
1243
+ "codebook_utilization": 0.960205078125,
1244
+ "lr": 5.522642316338268e-05,
1245
+ "phase": "full_vq",
1246
+ "val_recon_loss": 0.21258167947611112,
1247
+ "val_utilization": 0.948974609375,
1248
+ "time_sec": 141.5059130191803
1249
+ },
1250
+ {
1251
+ "epoch": 96,
1252
+ "loss": 0.2583759027687192,
1253
+ "recon_loss": 0.20922333516967975,
1254
+ "commit_loss": 0.024576283991336823,
1255
+ "embed_loss": 0.024576283991336823,
1256
+ "codebook_utilization": 0.9599609375,
1257
+ "lr": 5.4424655795567935e-05,
1258
+ "phase": "full_vq",
1259
+ "val_recon_loss": 0.2125933262404431,
1260
+ "val_utilization": 0.948974609375,
1261
+ "time_sec": 139.16457152366638
1262
+ },
1263
+ {
1264
+ "epoch": 97,
1265
+ "loss": 0.25822049238009825,
1266
+ "recon_loss": 0.20917262332791559,
1267
+ "commit_loss": 0.024523934104598473,
1268
+ "embed_loss": 0.024523934104598473,
1269
+ "codebook_utilization": 0.960205078125,
1270
+ "lr": 5.3621740008088126e-05,
1271
+ "phase": "full_vq",
1272
+ "val_recon_loss": 0.2125886579075556,
1273
+ "val_utilization": 0.949462890625,
1274
+ "time_sec": 140.99408268928528
1275
+ },
1276
+ {
1277
+ "epoch": 98,
1278
+ "loss": 0.2582162002399362,
1279
+ "recon_loss": 0.20907579077695443,
1280
+ "commit_loss": 0.02457020467596487,
1281
+ "embed_loss": 0.02457020467596487,
1282
+ "codebook_utilization": 0.960205078125,
1283
+ "lr": 5.281788419778186e-05,
1284
+ "phase": "full_vq",
1285
+ "val_recon_loss": 0.21248957941706262,
1286
+ "val_utilization": 0.951904296875,
1287
+ "time_sec": 140.97324514389038
1288
+ },
1289
+ {
1290
+ "epoch": 99,
1291
+ "loss": 0.2582078810710571,
1292
+ "recon_loss": 0.20913958064908902,
1293
+ "commit_loss": 0.024534150198364486,
1294
+ "embed_loss": 0.024534150198364486,
1295
+ "codebook_utilization": 0.960205078125,
1296
+ "lr": 5.201329700547076e-05,
1297
+ "phase": "full_vq",
1298
+ "val_recon_loss": 0.2126225953021746,
1299
+ "val_utilization": 0.95068359375,
1300
+ "time_sec": 140.8686535358429
1301
+ },
1302
+ {
1303
+ "epoch": 100,
1304
+ "loss": 0.25852710026793363,
1305
+ "recon_loss": 0.20913594236299599,
1306
+ "commit_loss": 0.02469557864707579,
1307
+ "embed_loss": 0.02469557864707579,
1308
+ "codebook_utilization": 0.960205078125,
1309
+ "lr": 5.120818726180663e-05,
1310
+ "phase": "full_vq",
1311
+ "val_recon_loss": 0.21254579814967145,
1312
+ "val_utilization": 0.947265625,
1313
+ "time_sec": 141.5156168937683
1314
+ },
1315
+ {
1316
+ "epoch": 101,
1317
+ "loss": 0.2582454721173297,
1318
+ "recon_loss": 0.20911990897603797,
1319
+ "commit_loss": 0.02456278161607622,
1320
+ "embed_loss": 0.02456278161607622,
1321
+ "codebook_utilization": 0.960205078125,
1322
+ "lr": 5.040276393306951e-05,
1323
+ "phase": "full_vq",
1324
+ "val_recon_loss": 0.2125664883282747,
1325
+ "val_utilization": 0.949951171875,
1326
+ "time_sec": 143.38250136375427
1327
+ },
1328
+ {
1329
+ "epoch": 102,
1330
+ "loss": 0.2584769283408718,
1331
+ "recon_loss": 0.20910802336162346,
1332
+ "commit_loss": 0.024684452383619982,
1333
+ "embed_loss": 0.024684452383619982,
1334
+ "codebook_utilization": 0.960205078125,
1335
+ "lr": 4.9597236066930516e-05,
1336
+ "phase": "full_vq",
1337
+ "val_recon_loss": 0.21264461720927377,
1338
+ "val_utilization": 0.94921875,
1339
+ "time_sec": 146.97553181648254
1340
+ },
1341
+ {
1342
+ "epoch": 103,
1343
+ "loss": 0.25829327730386237,
1344
+ "recon_loss": 0.20895708715770303,
1345
+ "commit_loss": 0.024668095085699223,
1346
+ "embed_loss": 0.024668095085699223,
1347
+ "codebook_utilization": 0.960205078125,
1348
+ "lr": 4.879181273819339e-05,
1349
+ "phase": "full_vq",
1350
+ "val_recon_loss": 0.21238811556877715,
1351
+ "val_utilization": 0.94775390625,
1352
+ "time_sec": 137.14095425605774
1353
+ },
1354
+ {
1355
+ "epoch": 104,
1356
+ "loss": 0.2583693628550222,
1357
+ "recon_loss": 0.20888261995745222,
1358
+ "commit_loss": 0.02474337125949178,
1359
+ "embed_loss": 0.02474337125949178,
1360
+ "codebook_utilization": 0.960205078125,
1361
+ "lr": 4.798670299452925e-05,
1362
+ "phase": "full_vq",
1363
+ "val_recon_loss": 0.21236943344721634,
1364
+ "val_utilization": 0.9462890625,
1365
+ "time_sec": 142.7677538394928
1366
+ },
1367
+ {
1368
+ "epoch": 105,
1369
+ "loss": 0.2583082742524664,
1370
+ "recon_loss": 0.20893791069587073,
1371
+ "commit_loss": 0.024685181843919483,
1372
+ "embed_loss": 0.024685181843919483,
1373
+ "codebook_utilization": 0.960205078125,
1374
+ "lr": 4.718211580221814e-05,
1375
+ "phase": "full_vq",
1376
+ "val_recon_loss": 0.21264000061187852,
1377
+ "val_utilization": 0.950927734375,
1378
+ "time_sec": 143.27975416183472
1379
+ },
1380
+ {
1381
+ "epoch": 106,
1382
+ "loss": 0.25840118254830197,
1383
+ "recon_loss": 0.2088395162488064,
1384
+ "commit_loss": 0.024780833210321623,
1385
+ "embed_loss": 0.024780833210321623,
1386
+ "codebook_utilization": 0.960205078125,
1387
+ "lr": 4.637825999191189e-05,
1388
+ "phase": "full_vq",
1389
+ "val_recon_loss": 0.2125720442848259,
1390
+ "val_utilization": 0.949951171875,
1391
+ "time_sec": 146.38738584518433
1392
+ },
1393
+ {
1394
+ "epoch": 107,
1395
+ "loss": 0.25836430901918955,
1396
+ "recon_loss": 0.20888379725014292,
1397
+ "commit_loss": 0.024740255889571135,
1398
+ "embed_loss": 0.024740255889571135,
1399
+ "codebook_utilization": 0.959716796875,
1400
+ "lr": 4.557534420443209e-05,
1401
+ "phase": "full_vq",
1402
+ "val_recon_loss": 0.21220649469099687,
1403
+ "val_utilization": 0.94970703125,
1404
+ "time_sec": 144.89909029006958
1405
+ },
1406
+ {
1407
+ "epoch": 108,
1408
+ "loss": 0.2585114060742099,
1409
+ "recon_loss": 0.20884302870287158,
1410
+ "commit_loss": 0.02483418873362344,
1411
+ "embed_loss": 0.02483418873362344,
1412
+ "codebook_utilization": 0.9599609375,
1413
+ "lr": 4.477357683661733e-05,
1414
+ "phase": "full_vq",
1415
+ "val_recon_loss": 0.2123807014876537,
1416
+ "val_utilization": 0.94873046875,
1417
+ "time_sec": 140.34827756881714
1418
+ },
1419
+ {
1420
+ "epoch": 109,
1421
+ "loss": 0.2582180574132498,
1422
+ "recon_loss": 0.20870091235088462,
1423
+ "commit_loss": 0.02475857282647999,
1424
+ "embed_loss": 0.02475857282647999,
1425
+ "codebook_utilization": 0.960205078125,
1426
+ "lr": 4.397316598723385e-05,
1427
+ "phase": "full_vq",
1428
+ "val_recon_loss": 0.21223998848307,
1429
+ "val_utilization": 0.9482421875,
1430
+ "time_sec": 143.87352299690247
1431
+ },
1432
+ {
1433
+ "epoch": 110,
1434
+ "loss": 0.2583341572867823,
1435
+ "recon_loss": 0.20869129758863267,
1436
+ "commit_loss": 0.02482142974054668,
1437
+ "embed_loss": 0.02482142974054668,
1438
+ "codebook_utilization": 0.9599609375,
1439
+ "lr": 4.317431940296343e-05,
1440
+ "phase": "full_vq",
1441
+ "val_recon_loss": 0.21213393030541666,
1442
+ "val_utilization": 0.94970703125,
1443
+ "time_sec": 141.30009245872498
1444
+ },
1445
+ {
1446
+ "epoch": 111,
1447
+ "loss": 0.25840904939223114,
1448
+ "recon_loss": 0.20866976092824444,
1449
+ "commit_loss": 0.024869644466716908,
1450
+ "embed_loss": 0.024869644466716908,
1451
+ "codebook_utilization": 0.960205078125,
1452
+ "lr": 4.237724442448274e-05,
1453
+ "phase": "full_vq",
1454
+ "val_recon_loss": 0.21240845029608588,
1455
+ "val_utilization": 0.94921875,
1456
+ "time_sec": 139.70933651924133
1457
+ },
1458
+ {
1459
+ "epoch": 112,
1460
+ "loss": 0.25831283544137223,
1461
+ "recon_loss": 0.20862274271685902,
1462
+ "commit_loss": 0.02484504613258086,
1463
+ "embed_loss": 0.02484504613258086,
1464
+ "codebook_utilization": 0.960205078125,
1465
+ "lr": 4.158214793264808e-05,
1466
+ "phase": "full_vq",
1467
+ "val_recon_loss": 0.21250487269645327,
1468
+ "val_utilization": 0.94873046875,
1469
+ "time_sec": 141.85187792778015
1470
+ },
1471
+ {
1472
+ "epoch": 113,
1473
+ "loss": 0.25846623372902394,
1474
+ "recon_loss": 0.20870393262242237,
1475
+ "commit_loss": 0.024881150432153123,
1476
+ "embed_loss": 0.024881150432153123,
1477
+ "codebook_utilization": 0.960205078125,
1478
+ "lr": 4.078923629479945e-05,
1479
+ "phase": "full_vq",
1480
+ "val_recon_loss": 0.21218352689501946,
1481
+ "val_utilization": 0.9501953125,
1482
+ "time_sec": 143.94795060157776
1483
+ },
1484
+ {
1485
+ "epoch": 114,
1486
+ "loss": 0.2583673260803145,
1487
+ "recon_loss": 0.20856938190091917,
1488
+ "commit_loss": 0.024898972180558414,
1489
+ "embed_loss": 0.024898972180558414,
1490
+ "codebook_utilization": 0.960205078125,
1491
+ "lr": 3.9998715311197785e-05,
1492
+ "phase": "full_vq",
1493
+ "val_recon_loss": 0.21241962767384026,
1494
+ "val_utilization": 0.95068359375,
1495
+ "time_sec": 141.1611099243164
1496
+ },
1497
+ {
1498
+ "epoch": 115,
1499
+ "loss": 0.25848160659879205,
1500
+ "recon_loss": 0.20859493636261156,
1501
+ "commit_loss": 0.024943335072659865,
1502
+ "embed_loss": 0.024943335072659865,
1503
+ "codebook_utilization": 0.9599609375,
1504
+ "lr": 3.9210790161609705e-05,
1505
+ "phase": "full_vq",
1506
+ "val_recon_loss": 0.21232353092244502,
1507
+ "val_utilization": 0.9501953125,
1508
+ "time_sec": 144.22900700569153
1509
+ },
1510
+ {
1511
+ "epoch": 116,
1512
+ "loss": 0.2584174014123152,
1513
+ "recon_loss": 0.20851436788392907,
1514
+ "commit_loss": 0.024951516595091115,
1515
+ "embed_loss": 0.024951516595091115,
1516
+ "codebook_utilization": 0.9599609375,
1517
+ "lr": 3.8425665352052864e-05,
1518
+ "phase": "full_vq",
1519
+ "val_recon_loss": 0.21215088791056966,
1520
+ "val_utilization": 0.951904296875,
1521
+ "time_sec": 146.63667154312134
1522
+ },
1523
+ {
1524
+ "epoch": 117,
1525
+ "loss": 0.2583816125585135,
1526
+ "recon_loss": 0.20847267491949928,
1527
+ "commit_loss": 0.024954468589831337,
1528
+ "embed_loss": 0.024954468589831337,
1529
+ "codebook_utilization": 0.9599609375,
1530
+ "lr": 3.764354466171653e-05,
1531
+ "phase": "full_vq",
1532
+ "val_recon_loss": 0.21224491792113592,
1533
+ "val_utilization": 0.950439453125,
1534
+ "time_sec": 146.19459009170532
1535
+ },
1536
+ {
1537
+ "epoch": 118,
1538
+ "loss": 0.2585952648423552,
1539
+ "recon_loss": 0.20857626351156855,
1540
+ "commit_loss": 0.02500950098088198,
1541
+ "embed_loss": 0.02500950098088198,
1542
+ "codebook_utilization": 0.9599609375,
1543
+ "lr": 3.686463109007067e-05,
1544
+ "phase": "full_vq",
1545
+ "val_recon_loss": 0.21221903100442352,
1546
+ "val_utilization": 0.948974609375,
1547
+ "time_sec": 146.83803415298462
1548
+ },
1549
+ {
1550
+ "epoch": 119,
1551
+ "loss": 0.25845359051776773,
1552
+ "recon_loss": 0.20843913270852107,
1553
+ "commit_loss": 0.02500722866737584,
1554
+ "embed_loss": 0.02500722866737584,
1555
+ "codebook_utilization": 0.960205078125,
1556
+ "lr": 3.6089126804177374e-05,
1557
+ "phase": "full_vq",
1558
+ "val_recon_loss": 0.21204945512032242,
1559
+ "val_utilization": 0.950927734375,
1560
+ "time_sec": 143.42903184890747
1561
+ },
1562
+ {
1563
+ "epoch": 120,
1564
+ "loss": 0.2585765772920637,
1565
+ "recon_loss": 0.20844616718731598,
1566
+ "commit_loss": 0.02506520497918048,
1567
+ "embed_loss": 0.02506520497918048,
1568
+ "codebook_utilization": 0.959716796875,
1569
+ "lr": 3.531723308621847e-05,
1570
+ "phase": "full_vq",
1571
+ "val_recon_loss": 0.21198806844735413,
1572
+ "val_utilization": 0.9501953125,
1573
+ "time_sec": 143.7408845424652
1574
+ },
1575
+ {
1576
+ "epoch": 121,
1577
+ "loss": 0.2585934158868906,
1578
+ "recon_loss": 0.2083775931136395,
1579
+ "commit_loss": 0.025107911134234613,
1580
+ "embed_loss": 0.025107911134234613,
1581
+ "codebook_utilization": 0.960205078125,
1582
+ "lr": 3.4549150281252636e-05,
1583
+ "phase": "full_vq",
1584
+ "val_recon_loss": 0.21209996620590768,
1585
+ "val_utilization": 0.948486328125,
1586
+ "time_sec": 145.12657403945923
1587
+ },
1588
+ {
1589
+ "epoch": 122,
1590
+ "loss": 0.2583266192981544,
1591
+ "recon_loss": 0.2083003449084636,
1592
+ "commit_loss": 0.025013137391710346,
1593
+ "embed_loss": 0.025013137391710346,
1594
+ "codebook_utilization": 0.960205078125,
1595
+ "lr": 3.3785077745215873e-05,
1596
+ "phase": "full_vq",
1597
+ "val_recon_loss": 0.21192585785737197,
1598
+ "val_utilization": 0.952392578125,
1599
+ "time_sec": 144.582373380661
1600
+ },
1601
+ {
1602
+ "epoch": 123,
1603
+ "loss": 0.2584562777342189,
1604
+ "recon_loss": 0.20828300076448497,
1605
+ "commit_loss": 0.025086638727162266,
1606
+ "embed_loss": 0.025086638727162266,
1607
+ "codebook_utilization": 0.960205078125,
1608
+ "lr": 3.302521379317865e-05,
1609
+ "phase": "full_vq",
1610
+ "val_recon_loss": 0.21207065344526527,
1611
+ "val_utilization": 0.9482421875,
1612
+ "time_sec": 141.4992265701294
1613
+ },
1614
+ {
1615
+ "epoch": 124,
1616
+ "loss": 0.2585316289085037,
1617
+ "recon_loss": 0.20828487966361084,
1618
+ "commit_loss": 0.025123374652733326,
1619
+ "embed_loss": 0.025123374652733326,
1620
+ "codebook_utilization": 0.959716796875,
1621
+ "lr": 3.226975564787323e-05,
1622
+ "phase": "full_vq",
1623
+ "val_recon_loss": 0.21218894932712062,
1624
+ "val_utilization": 0.94677734375,
1625
+ "time_sec": 143.4921097755432
1626
+ },
1627
+ {
1628
+ "epoch": 125,
1629
+ "loss": 0.2585887834429741,
1630
+ "recon_loss": 0.20830878807196448,
1631
+ "commit_loss": 0.025139997829367638,
1632
+ "embed_loss": 0.025139997829367638,
1633
+ "codebook_utilization": 0.960205078125,
1634
+ "lr": 3.151889938850445e-05,
1635
+ "phase": "full_vq",
1636
+ "val_recon_loss": 0.2121136537596081,
1637
+ "val_utilization": 0.94921875,
1638
+ "time_sec": 140.8396110534668
1639
+ },
1640
+ {
1641
+ "epoch": 126,
1642
+ "loss": 0.25860217211611547,
1643
+ "recon_loss": 0.20821322969143308,
1644
+ "commit_loss": 0.02519447130320195,
1645
+ "embed_loss": 0.02519447130320195,
1646
+ "codebook_utilization": 0.9599609375,
1647
+ "lr": 3.0772839899857464e-05,
1648
+ "phase": "full_vq",
1649
+ "val_recon_loss": 0.21194713939441723,
1650
+ "val_utilization": 0.94921875,
1651
+ "time_sec": 138.5580816268921
1652
+ },
1653
+ {
1654
+ "epoch": 127,
1655
+ "loss": 0.2585419769371105,
1656
+ "recon_loss": 0.20819944803549992,
1657
+ "commit_loss": 0.025171264314514188,
1658
+ "embed_loss": 0.025171264314514188,
1659
+ "codebook_utilization": 0.960205078125,
1660
+ "lr": 3.003177082171522e-05,
1661
+ "phase": "full_vq",
1662
+ "val_recon_loss": 0.21203415020463173,
1663
+ "val_utilization": 0.947998046875,
1664
+ "time_sec": 142.41585659980774
1665
+ },
1666
+ {
1667
+ "epoch": 128,
1668
+ "loss": 0.25851718059723294,
1669
+ "recon_loss": 0.20819601753782127,
1670
+ "commit_loss": 0.02516058168871212,
1671
+ "embed_loss": 0.02516058168871212,
1672
+ "codebook_utilization": 0.9599609375,
1673
+ "lr": 2.92958844985994e-05,
1674
+ "phase": "full_vq",
1675
+ "val_recon_loss": 0.21195581170280328,
1676
+ "val_utilization": 0.951171875,
1677
+ "time_sec": 141.58988904953003
1678
+ },
1679
+ {
1680
+ "epoch": 129,
1681
+ "loss": 0.25869512133966616,
1682
+ "recon_loss": 0.2081550459672765,
1683
+ "commit_loss": 0.02527003786286849,
1684
+ "embed_loss": 0.02527003786286849,
1685
+ "codebook_utilization": 0.960205078125,
1686
+ "lr": 2.8565371929847304e-05,
1687
+ "phase": "full_vq",
1688
+ "val_recon_loss": 0.2121603623702285,
1689
+ "val_utilization": 0.94970703125,
1690
+ "time_sec": 144.01764297485352
1691
+ },
1692
+ {
1693
+ "epoch": 130,
1694
+ "loss": 0.25863794501880966,
1695
+ "recon_loss": 0.20815599437167004,
1696
+ "commit_loss": 0.025240975611295482,
1697
+ "embed_loss": 0.025240975611295482,
1698
+ "codebook_utilization": 0.960205078125,
1699
+ "lr": 2.7840422720037937e-05,
1700
+ "phase": "full_vq",
1701
+ "val_recon_loss": 0.2118921877627962,
1702
+ "val_utilization": 0.948974609375,
1703
+ "time_sec": 148.99665331840515
1704
+ },
1705
+ {
1706
+ "epoch": 131,
1707
+ "loss": 0.2585354602231889,
1708
+ "recon_loss": 0.20807598220300544,
1709
+ "commit_loss": 0.02522973912619155,
1710
+ "embed_loss": 0.02522973912619155,
1711
+ "codebook_utilization": 0.9599609375,
1712
+ "lr": 2.7121225029780234e-05,
1713
+ "phase": "full_vq",
1714
+ "val_recon_loss": 0.21195950292134552,
1715
+ "val_utilization": 0.950439453125,
1716
+ "time_sec": 146.20467519760132
1717
+ },
1718
+ {
1719
+ "epoch": 132,
1720
+ "loss": 0.2586112513855544,
1721
+ "recon_loss": 0.20804637535235423,
1722
+ "commit_loss": 0.025282437991360984,
1723
+ "embed_loss": 0.025282437991360984,
1724
+ "codebook_utilization": 0.960205078125,
1725
+ "lr": 2.6407965526875892e-05,
1726
+ "phase": "full_vq",
1727
+ "val_recon_loss": 0.2120183923103836,
1728
+ "val_utilization": 0.947998046875,
1729
+ "time_sec": 143.6440305709839
1730
+ },
1731
+ {
1732
+ "epoch": 133,
1733
+ "loss": 0.2585967746247767,
1734
+ "recon_loss": 0.2080702587033352,
1735
+ "commit_loss": 0.02526325788247957,
1736
+ "embed_loss": 0.02526325788247957,
1737
+ "codebook_utilization": 0.960205078125,
1738
+ "lr": 2.570082933786969e-05,
1739
+ "phase": "full_vq",
1740
+ "val_recon_loss": 0.21198202326391521,
1741
+ "val_utilization": 0.949951171875,
1742
+ "time_sec": 143.77961444854736
1743
+ },
1744
+ {
1745
+ "epoch": 134,
1746
+ "loss": 0.25867608476298937,
1747
+ "recon_loss": 0.2080024917960813,
1748
+ "commit_loss": 0.02533679651626484,
1749
+ "embed_loss": 0.02533679651626484,
1750
+ "codebook_utilization": 0.9599609375,
1751
+ "lr": 2.5000000000000008e-05,
1752
+ "phase": "full_vq",
1753
+ "val_recon_loss": 0.21196440004565742,
1754
+ "val_utilization": 0.949951171875,
1755
+ "time_sec": 142.69802403450012
1756
+ },
1757
+ {
1758
+ "epoch": 135,
1759
+ "loss": 0.2587372580477539,
1760
+ "recon_loss": 0.20803141696908609,
1761
+ "commit_loss": 0.025352920352564593,
1762
+ "embed_loss": 0.025352920352564593,
1763
+ "codebook_utilization": 0.9599609375,
1764
+ "lr": 2.4305659413561568e-05,
1765
+ "phase": "full_vq",
1766
+ "val_recon_loss": 0.21198264316896373,
1767
+ "val_utilization": 0.948974609375,
1768
+ "time_sec": 135.50866651535034
1769
+ },
1770
+ {
1771
+ "epoch": 136,
1772
+ "loss": 0.2585985584633783,
1773
+ "recon_loss": 0.20796804077580047,
1774
+ "commit_loss": 0.025315258740308648,
1775
+ "embed_loss": 0.025315258740308648,
1776
+ "codebook_utilization": 0.9599609375,
1777
+ "lr": 2.3617987794693356e-05,
1778
+ "phase": "full_vq",
1779
+ "val_recon_loss": 0.21199960403897788,
1780
+ "val_utilization": 0.9501953125,
1781
+ "time_sec": 136.75576210021973
1782
+ },
1783
+ {
1784
+ "epoch": 137,
1785
+ "loss": 0.25866809953761294,
1786
+ "recon_loss": 0.2079689533406803,
1787
+ "commit_loss": 0.025349573032844682,
1788
+ "embed_loss": 0.025349573032844682,
1789
+ "codebook_utilization": 0.9599609375,
1790
+ "lr": 2.2937163628603435e-05,
1791
+ "phase": "full_vq",
1792
+ "val_recon_loss": 0.21196182247962844,
1793
+ "val_utilization": 0.951171875,
1794
+ "time_sec": 138.0742015838623
1795
+ },
1796
+ {
1797
+ "epoch": 138,
1798
+ "loss": 0.25864017700276726,
1799
+ "recon_loss": 0.20790566724086518,
1800
+ "commit_loss": 0.0253672551156746,
1801
+ "embed_loss": 0.0253672551156746,
1802
+ "codebook_utilization": 0.960205078125,
1803
+ "lr": 2.2263363623243054e-05,
1804
+ "phase": "full_vq",
1805
+ "val_recon_loss": 0.21189745582556457,
1806
+ "val_utilization": 0.94775390625,
1807
+ "time_sec": 139.20591282844543
1808
+ },
1809
+ {
1810
+ "epoch": 139,
1811
+ "loss": 0.258666986392604,
1812
+ "recon_loss": 0.20795427420274998,
1813
+ "commit_loss": 0.025356355935920706,
1814
+ "embed_loss": 0.025356355935920706,
1815
+ "codebook_utilization": 0.959716796875,
1816
+ "lr": 2.1596762663442214e-05,
1817
+ "phase": "full_vq",
1818
+ "val_recon_loss": 0.2118665027651894,
1819
+ "val_utilization": 0.94921875,
1820
+ "time_sec": 145.27109384536743
1821
+ },
1822
+ {
1823
+ "epoch": 140,
1824
+ "loss": 0.258794684780807,
1825
+ "recon_loss": 0.20794706523095366,
1826
+ "commit_loss": 0.025423809646207305,
1827
+ "embed_loss": 0.025423809646207305,
1828
+ "codebook_utilization": 0.959716796875,
1829
+ "lr": 2.0937533765518167e-05,
1830
+ "phase": "full_vq",
1831
+ "val_recon_loss": 0.2118346607417203,
1832
+ "val_utilization": 0.948486328125,
1833
+ "time_sec": 136.94103169441223
1834
+ },
1835
+ {
1836
+ "epoch": 141,
1837
+ "loss": 0.25881551135442443,
1838
+ "recon_loss": 0.2079573765922045,
1839
+ "commit_loss": 0.025429067446731617,
1840
+ "embed_loss": 0.025429067446731617,
1841
+ "codebook_utilization": 0.959716796875,
1842
+ "lr": 2.0285848032369133e-05,
1843
+ "phase": "full_vq",
1844
+ "val_recon_loss": 0.21194372259164124,
1845
+ "val_utilization": 0.949951171875,
1846
+ "time_sec": 141.69983077049255
1847
+ },
1848
+ {
1849
+ "epoch": 142,
1850
+ "loss": 0.2588034980100022,
1851
+ "recon_loss": 0.20790888623493473,
1852
+ "commit_loss": 0.02544730563514278,
1853
+ "embed_loss": 0.02544730563514278,
1854
+ "codebook_utilization": 0.959716796875,
1855
+ "lr": 1.964187460906444e-05,
1856
+ "phase": "full_vq",
1857
+ "val_recon_loss": 0.21174719762266353,
1858
+ "val_utilization": 0.9482421875,
1859
+ "time_sec": 144.94414067268372
1860
+ },
1861
+ {
1862
+ "epoch": 143,
1863
+ "loss": 0.2587182930289568,
1864
+ "recon_loss": 0.20785080181147025,
1865
+ "commit_loss": 0.025433745416926175,
1866
+ "embed_loss": 0.025433745416926175,
1867
+ "codebook_utilization": 0.959716796875,
1868
+ "lr": 1.9005780638942995e-05,
1869
+ "phase": "full_vq",
1870
+ "val_recon_loss": 0.2118924908925978,
1871
+ "val_utilization": 0.947509765625,
1872
+ "time_sec": 141.66879534721375
1873
+ },
1874
+ {
1875
+ "epoch": 144,
1876
+ "loss": 0.25872288880551736,
1877
+ "recon_loss": 0.20778209358696045,
1878
+ "commit_loss": 0.02547039743512869,
1879
+ "embed_loss": 0.02547039743512869,
1880
+ "codebook_utilization": 0.95947265625,
1881
+ "lr": 1.837773122023114e-05,
1882
+ "phase": "full_vq",
1883
+ "val_recon_loss": 0.21197703380263253,
1884
+ "val_utilization": 0.94873046875,
1885
+ "time_sec": 140.22795939445496
1886
+ },
1887
+ {
1888
+ "epoch": 145,
1889
+ "loss": 0.25878342640836066,
1890
+ "recon_loss": 0.20782248811104756,
1891
+ "commit_loss": 0.02548046927485201,
1892
+ "embed_loss": 0.02548046927485201,
1893
+ "codebook_utilization": 0.959716796875,
1894
+ "lr": 1.775788936319148e-05,
1895
+ "phase": "full_vq",
1896
+ "val_recon_loss": 0.21184307077292647,
1897
+ "val_utilization": 0.9501953125,
1898
+ "time_sec": 142.39927697181702
1899
+ },
1900
+ {
1901
+ "epoch": 146,
1902
+ "loss": 0.2588493183537873,
1903
+ "recon_loss": 0.20780124198292005,
1904
+ "commit_loss": 0.02552403837220294,
1905
+ "embed_loss": 0.02552403837220294,
1906
+ "codebook_utilization": 0.95947265625,
1907
+ "lr": 1.714641594781347e-05,
1908
+ "phase": "full_vq",
1909
+ "val_recon_loss": 0.21183166485489086,
1910
+ "val_utilization": 0.9482421875,
1911
+ "time_sec": 145.58855938911438
1912
+ },
1913
+ {
1914
+ "epoch": 147,
1915
+ "loss": 0.25880129504090726,
1916
+ "recon_loss": 0.2078126541119281,
1917
+ "commit_loss": 0.025494320578065506,
1918
+ "embed_loss": 0.025494320578065506,
1919
+ "codebook_utilization": 0.95947265625,
1920
+ "lr": 1.654346968205709e-05,
1921
+ "phase": "full_vq",
1922
+ "val_recon_loss": 0.21173519482103625,
1923
+ "val_utilization": 0.948486328125,
1924
+ "time_sec": 142.40415167808533
1925
+ },
1926
+ {
1927
+ "epoch": 148,
1928
+ "loss": 0.25887914211930946,
1929
+ "recon_loss": 0.2077453019940433,
1930
+ "commit_loss": 0.025566919885959407,
1931
+ "embed_loss": 0.025566919885959407,
1932
+ "codebook_utilization": 0.959716796875,
1933
+ "lr": 1.5949207060660148e-05,
1934
+ "phase": "full_vq",
1935
+ "val_recon_loss": 0.2119110973363512,
1936
+ "val_utilization": 0.9482421875,
1937
+ "time_sec": 142.70816850662231
1938
+ },
1939
+ {
1940
+ "epoch": 149,
1941
+ "loss": 0.25884581514732624,
1942
+ "recon_loss": 0.20773883711112548,
1943
+ "commit_loss": 0.0255534889676222,
1944
+ "embed_loss": 0.0255534889676222,
1945
+ "codebook_utilization": 0.959716796875,
1946
+ "lr": 1.536378232452003e-05,
1947
+ "phase": "full_vq",
1948
+ "val_recon_loss": 0.2117347943732578,
1949
+ "val_utilization": 0.9482421875,
1950
+ "time_sec": 143.41825151443481
1951
+ },
1952
+ {
1953
+ "epoch": 150,
1954
+ "loss": 0.25883854368353276,
1955
+ "recon_loss": 0.20773012764489424,
1956
+ "commit_loss": 0.025554208198516833,
1957
+ "embed_loss": 0.025554208198516833,
1958
+ "codebook_utilization": 0.959716796875,
1959
+ "lr": 1.478734742066054e-05,
1960
+ "phase": "full_vq",
1961
+ "val_recon_loss": 0.21178821415713664,
1962
+ "val_utilization": 0.95068359375,
1963
+ "time_sec": 142.56376266479492
1964
+ },
1965
+ {
1966
+ "epoch": 151,
1967
+ "loss": 0.2589412463228231,
1968
+ "recon_loss": 0.20769854408857946,
1969
+ "commit_loss": 0.025621350925304702,
1970
+ "embed_loss": 0.025621350925304702,
1971
+ "codebook_utilization": 0.959716796875,
1972
+ "lr": 1.4220051962793949e-05,
1973
+ "phase": "full_vq",
1974
+ "val_recon_loss": 0.21184555675541417,
1975
+ "val_utilization": 0.94921875,
1976
+ "time_sec": 146.39625930786133
1977
+ },
1978
+ {
1979
+ "epoch": 152,
1980
+ "loss": 0.25891455093292687,
1981
+ "recon_loss": 0.20766843594913562,
1982
+ "commit_loss": 0.025623057648378012,
1983
+ "embed_loss": 0.025623057648378012,
1984
+ "codebook_utilization": 0.959716796875,
1985
+ "lr": 1.3662043192488835e-05,
1986
+ "phase": "full_vq",
1987
+ "val_recon_loss": 0.21177927011184478,
1988
+ "val_utilization": 0.947998046875,
1989
+ "time_sec": 141.51404666900635
1990
+ },
1991
+ {
1992
+ "epoch": 153,
1993
+ "loss": 0.2588522950929355,
1994
+ "recon_loss": 0.20767886637833707,
1995
+ "commit_loss": 0.025586714541544597,
1996
+ "embed_loss": 0.025586714541544597,
1997
+ "codebook_utilization": 0.959716796875,
1998
+ "lr": 1.3113465940953492e-05,
1999
+ "phase": "full_vq",
2000
+ "val_recon_loss": 0.211817874918493,
2001
+ "val_utilization": 0.95068359375,
2002
+ "time_sec": 140.0248987674713
2003
+ },
2004
+ {
2005
+ "epoch": 154,
2006
+ "loss": 0.25899658923549707,
2007
+ "recon_loss": 0.20765761128528332,
2008
+ "commit_loss": 0.02566948889938958,
2009
+ "embed_loss": 0.02566948889938958,
2010
+ "codebook_utilization": 0.959716796875,
2011
+ "lr": 1.2574462591444937e-05,
2012
+ "phase": "full_vq",
2013
+ "val_recon_loss": 0.21184951761800253,
2014
+ "val_utilization": 0.950439453125,
2015
+ "time_sec": 140.22791051864624
2016
+ },
2017
+ {
2018
+ "epoch": 155,
2019
+ "loss": 0.25901680902090823,
2020
+ "recon_loss": 0.2077227641694591,
2021
+ "commit_loss": 0.025647022329816003,
2022
+ "embed_loss": 0.025647022329816003,
2023
+ "codebook_utilization": 0.959716796875,
2024
+ "lr": 1.2045173042313425e-05,
2025
+ "phase": "full_vq",
2026
+ "val_recon_loss": 0.211779128885671,
2027
+ "val_utilization": 0.9501953125,
2028
+ "time_sec": 143.81999945640564
2029
+ },
2030
+ {
2031
+ "epoch": 156,
2032
+ "loss": 0.2589520044362319,
2033
+ "recon_loss": 0.20762247679033252,
2034
+ "commit_loss": 0.025664763653847743,
2035
+ "embed_loss": 0.025664763653847743,
2036
+ "codebook_utilization": 0.95947265625,
2037
+ "lr": 1.1525734670691714e-05,
2038
+ "phase": "full_vq",
2039
+ "val_recon_loss": 0.21173930168151855,
2040
+ "val_utilization": 0.9501953125,
2041
+ "time_sec": 146.519535779953
2042
+ },
2043
+ {
2044
+ "epoch": 157,
2045
+ "loss": 0.2590464165580628,
2046
+ "recon_loss": 0.20760027074878454,
2047
+ "commit_loss": 0.025723072738061105,
2048
+ "embed_loss": 0.025723072738061105,
2049
+ "codebook_utilization": 0.959716796875,
2050
+ "lr": 1.1016282296838872e-05,
2051
+ "phase": "full_vq",
2052
+ "val_recon_loss": 0.21177494676595324,
2053
+ "val_utilization": 0.9482421875,
2054
+ "time_sec": 146.18688917160034
2055
+ },
2056
+ {
2057
+ "epoch": 158,
2058
+ "loss": 0.25892583208113185,
2059
+ "recon_loss": 0.2075854344217758,
2060
+ "commit_loss": 0.025670198918014846,
2061
+ "embed_loss": 0.025670198918014846,
2062
+ "codebook_utilization": 0.959716796875,
2063
+ "lr": 1.051694814914775e-05,
2064
+ "phase": "full_vq",
2065
+ "val_recon_loss": 0.21172436887628576,
2066
+ "val_utilization": 0.949462890625,
2067
+ "time_sec": 145.48429346084595
2068
+ },
2069
+ {
2070
+ "epoch": 159,
2071
+ "loss": 0.2591026104522268,
2072
+ "recon_loss": 0.20759570426976454,
2073
+ "commit_loss": 0.025753452826220654,
2074
+ "embed_loss": 0.025753452826220654,
2075
+ "codebook_utilization": 0.959716796875,
2076
+ "lr": 1.002786182982495e-05,
2077
+ "phase": "full_vq",
2078
+ "val_recon_loss": 0.21178416694315633,
2079
+ "val_utilization": 0.9482421875,
2080
+ "time_sec": 147.320454120636
2081
+ },
2082
+ {
2083
+ "epoch": 160,
2084
+ "loss": 0.25908089055601496,
2085
+ "recon_loss": 0.2076170422681948,
2086
+ "commit_loss": 0.025731924060621074,
2087
+ "embed_loss": 0.025731924060621074,
2088
+ "codebook_utilization": 0.959716796875,
2089
+ "lr": 9.549150281252631e-06,
2090
+ "phase": "full_vq",
2091
+ "val_recon_loss": 0.21173595101310966,
2092
+ "val_utilization": 0.94970703125,
2093
+ "time_sec": 140.90730333328247
2094
+ },
2095
+ {
2096
+ "epoch": 161,
2097
+ "loss": 0.25901386273020327,
2098
+ "recon_loss": 0.2075580828154313,
2099
+ "commit_loss": 0.02572789016687046,
2100
+ "embed_loss": 0.02572789016687046,
2101
+ "codebook_utilization": 0.959716796875,
2102
+ "lr": 9.080937753040662e-06,
2103
+ "phase": "full_vq",
2104
+ "val_recon_loss": 0.2117305052665512,
2105
+ "val_utilization": 0.950439453125,
2106
+ "time_sec": 148.84632539749146
2107
+ },
2108
+ {
2109
+ "epoch": 162,
2110
+ "loss": 0.25909019788590876,
2111
+ "recon_loss": 0.20760185152373017,
2112
+ "commit_loss": 0.025744172999367803,
2113
+ "embed_loss": 0.025744172999367803,
2114
+ "codebook_utilization": 0.959716796875,
2115
+ "lr": 8.62334576977751e-06,
2116
+ "phase": "full_vq",
2117
+ "val_recon_loss": 0.21173915133047638,
2118
+ "val_utilization": 0.94873046875,
2119
+ "time_sec": 143.76673245429993
2120
+ },
2121
+ {
2122
+ "epoch": 163,
2123
+ "loss": 0.2590920419954672,
2124
+ "recon_loss": 0.20753852780756912,
2125
+ "commit_loss": 0.02577675733372044,
2126
+ "embed_loss": 0.02577675733372044,
2127
+ "codebook_utilization": 0.95947265625,
2128
+ "lr": 8.176493099488661e-06,
2129
+ "phase": "full_vq",
2130
+ "val_recon_loss": 0.21170530602168502,
2131
+ "val_utilization": 0.9501953125,
2132
+ "time_sec": 140.07061791419983
2133
+ },
2134
+ {
2135
+ "epoch": 164,
2136
+ "loss": 0.2590435311238617,
2137
+ "recon_loss": 0.207552566114803,
2138
+ "commit_loss": 0.025745482401049073,
2139
+ "embed_loss": 0.025745482401049073,
2140
+ "codebook_utilization": 0.959716796875,
2141
+ "lr": 7.740495722810268e-06,
2142
+ "phase": "full_vq",
2143
+ "val_recon_loss": 0.21171206034971088,
2144
+ "val_utilization": 0.9501953125,
2145
+ "time_sec": 140.4784553050995
2146
+ },
2147
+ {
2148
+ "epoch": 165,
2149
+ "loss": 0.25905534790621865,
2150
+ "recon_loss": 0.2075229843820983,
2151
+ "commit_loss": 0.025766181895827373,
2152
+ "embed_loss": 0.025766181895827373,
2153
+ "codebook_utilization": 0.959716796875,
2154
+ "lr": 7.315466802886386e-06,
2155
+ "phase": "full_vq",
2156
+ "val_recon_loss": 0.21171357055728354,
2157
+ "val_utilization": 0.94873046875,
2158
+ "time_sec": 143.55909371376038
2159
+ },
2160
+ {
2161
+ "epoch": 166,
2162
+ "loss": 0.2590092028261523,
2163
+ "recon_loss": 0.20749325503097962,
2164
+ "commit_loss": 0.02575797357957376,
2165
+ "embed_loss": 0.02575797357957376,
2166
+ "codebook_utilization": 0.959716796875,
2167
+ "lr": 6.901516655997545e-06,
2168
+ "phase": "full_vq",
2169
+ "val_recon_loss": 0.21171397833007105,
2170
+ "val_utilization": 0.948974609375,
2171
+ "time_sec": 144.1349983215332
2172
+ },
2173
+ {
2174
+ "epoch": 167,
2175
+ "loss": 0.25912667055763206,
2176
+ "recon_loss": 0.20752859382125421,
2177
+ "commit_loss": 0.02579903830256727,
2178
+ "embed_loss": 0.02579903830256727,
2179
+ "codebook_utilization": 0.959716796875,
2180
+ "lr": 6.4987527229280395e-06,
2181
+ "phase": "full_vq",
2182
+ "val_recon_loss": 0.21170426197768596,
2183
+ "val_utilization": 0.949462890625,
2184
+ "time_sec": 144.63631916046143
2185
+ },
2186
+ {
2187
+ "epoch": 168,
2188
+ "loss": 0.2591653968381688,
2189
+ "recon_loss": 0.20749058859135078,
2190
+ "commit_loss": 0.025837404143600282,
2191
+ "embed_loss": 0.025837404143600282,
2192
+ "codebook_utilization": 0.959716796875,
2193
+ "lr": 6.107279541079777e-06,
2194
+ "phase": "full_vq",
2195
+ "val_recon_loss": 0.21173707060934452,
2196
+ "val_utilization": 0.94921875,
2197
+ "time_sec": 146.89997816085815
2198
+ },
2199
+ {
2200
+ "epoch": 169,
2201
+ "loss": 0.25926882180014277,
2202
+ "recon_loss": 0.20753031518804996,
2203
+ "commit_loss": 0.025869253374191964,
2204
+ "embed_loss": 0.025869253374191964,
2205
+ "codebook_utilization": 0.959716796875,
2206
+ "lr": 5.727198717339518e-06,
2207
+ "phase": "full_vq",
2208
+ "val_recon_loss": 0.2116849771543835,
2209
+ "val_utilization": 0.948974609375,
2210
+ "time_sec": 143.2497284412384
2211
+ },
2212
+ {
2213
+ "epoch": 170,
2214
+ "loss": 0.25918665005343394,
2215
+ "recon_loss": 0.20747281083973443,
2216
+ "commit_loss": 0.025856919415032636,
2217
+ "embed_loss": 0.025856919415032636,
2218
+ "codebook_utilization": 0.959716796875,
2219
+ "lr": 5.358608901706794e-06,
2220
+ "phase": "full_vq",
2221
+ "val_recon_loss": 0.211706154299586,
2222
+ "val_utilization": 0.94970703125,
2223
+ "time_sec": 142.96398901939392
2224
+ },
2225
+ {
2226
+ "epoch": 171,
2227
+ "loss": 0.259118976793315,
2228
+ "recon_loss": 0.20741655320171418,
2229
+ "commit_loss": 0.025851211452548744,
2230
+ "embed_loss": 0.025851211452548744,
2231
+ "codebook_utilization": 0.959716796875,
2232
+ "lr": 5.001605761689385e-06,
2233
+ "phase": "full_vq",
2234
+ "val_recon_loss": 0.21174049800199068,
2235
+ "val_utilization": 0.949462890625,
2236
+ "time_sec": 140.51101970672607
2237
+ },
2238
+ {
2239
+ "epoch": 172,
2240
+ "loss": 0.2593167613999953,
2241
+ "recon_loss": 0.20753264174794117,
2242
+ "commit_loss": 0.02589205986388573,
2243
+ "embed_loss": 0.02589205986388573,
2244
+ "codebook_utilization": 0.959716796875,
2245
+ "lr": 4.656281957472728e-06,
2246
+ "phase": "full_vq",
2247
+ "val_recon_loss": 0.21166708249222027,
2248
+ "val_utilization": 0.9501953125,
2249
+ "time_sec": 142.2528941631317
2250
+ },
2251
+ {
2252
+ "epoch": 173,
2253
+ "loss": 0.259227692192486,
2254
+ "recon_loss": 0.2074566307227786,
2255
+ "commit_loss": 0.02588553083076225,
2256
+ "embed_loss": 0.02588553083076225,
2257
+ "codebook_utilization": 0.959716796875,
2258
+ "lr": 4.322727117869949e-06,
2259
+ "phase": "full_vq",
2260
+ "val_recon_loss": 0.21168125713809152,
2261
+ "val_utilization": 0.949462890625,
2262
+ "time_sec": 139.901624917984
2263
+ },
2264
+ {
2265
+ "epoch": 174,
2266
+ "loss": 0.2591192353022131,
2267
+ "recon_loss": 0.20742643893088106,
2268
+ "commit_loss": 0.02584639836738749,
2269
+ "embed_loss": 0.02584639836738749,
2270
+ "codebook_utilization": 0.959716796875,
2271
+ "lr": 4.0010278170587865e-06,
2272
+ "phase": "full_vq",
2273
+ "val_recon_loss": 0.21165724299596936,
2274
+ "val_utilization": 0.949462890625,
2275
+ "time_sec": 144.58672308921814
2276
+ },
2277
+ {
2278
+ "epoch": 175,
2279
+ "loss": 0.2591961749039368,
2280
+ "recon_loss": 0.20743126609945686,
2281
+ "commit_loss": 0.025882454533483278,
2282
+ "embed_loss": 0.025882454533483278,
2283
+ "codebook_utilization": 0.959716796875,
2284
+ "lr": 3.6912675521111804e-06,
2285
+ "phase": "full_vq",
2286
+ "val_recon_loss": 0.21166603509964568,
2287
+ "val_utilization": 0.948486328125,
2288
+ "time_sec": 141.23290824890137
2289
+ },
2290
+ {
2291
+ "epoch": 176,
2292
+ "loss": 0.2592162422856018,
2293
+ "recon_loss": 0.2074340205124723,
2294
+ "commit_loss": 0.02589111120710124,
2295
+ "embed_loss": 0.02589111120710124,
2296
+ "codebook_utilization": 0.959716796875,
2297
+ "lr": 3.393526721321614e-06,
2298
+ "phase": "full_vq",
2299
+ "val_recon_loss": 0.21167665000042218,
2300
+ "val_utilization": 0.949462890625,
2301
+ "time_sec": 140.16413235664368
2302
+ },
2303
+ {
2304
+ "epoch": 177,
2305
+ "loss": 0.25919763012953245,
2306
+ "recon_loss": 0.2073852508694822,
2307
+ "commit_loss": 0.025906189677979405,
2308
+ "embed_loss": 0.025906189677979405,
2309
+ "codebook_utilization": 0.959716796875,
2310
+ "lr": 3.1078826033397826e-06,
2311
+ "phase": "full_vq",
2312
+ "val_recon_loss": 0.21169628822401668,
2313
+ "val_utilization": 0.94970703125,
2314
+ "time_sec": 141.4409658908844
2315
+ },
2316
+ {
2317
+ "epoch": 178,
2318
+ "loss": 0.25927791443620596,
2319
+ "recon_loss": 0.20742537789955373,
2320
+ "commit_loss": 0.02592626831123257,
2321
+ "embed_loss": 0.02592626831123257,
2322
+ "codebook_utilization": 0.959716796875,
2323
+ "lr": 2.834409337112846e-06,
2324
+ "phase": "full_vq",
2325
+ "val_recon_loss": 0.21166977813739454,
2326
+ "val_utilization": 0.948974609375,
2327
+ "time_sec": 141.2410626411438
2328
+ },
2329
+ {
2330
+ "epoch": 179,
2331
+ "loss": 0.25934389679816355,
2332
+ "recon_loss": 0.20742654279480136,
2333
+ "commit_loss": 0.025958676890629095,
2334
+ "embed_loss": 0.025958676890629095,
2335
+ "codebook_utilization": 0.959716796875,
2336
+ "lr": 2.57317790264273e-06,
2337
+ "phase": "full_vq",
2338
+ "val_recon_loss": 0.2116734470460522,
2339
+ "val_utilization": 0.94921875,
2340
+ "time_sec": 133.89122128486633
2341
+ },
2342
+ {
2343
+ "epoch": 180,
2344
+ "loss": 0.2592523172699662,
2345
+ "recon_loss": 0.20739064894636794,
2346
+ "commit_loss": 0.025930833919503824,
2347
+ "embed_loss": 0.025930833919503824,
2348
+ "codebook_utilization": 0.959716796875,
2349
+ "lr": 2.324256102563187e-06,
2350
+ "phase": "full_vq",
2351
+ "val_recon_loss": 0.2116392573362656,
2352
+ "val_utilization": 0.948974609375,
2353
+ "time_sec": 144.48794198036194
2354
+ },
2355
+ {
2356
+ "epoch": 181,
2357
+ "loss": 0.25929241088347704,
2358
+ "recon_loss": 0.20738233623869698,
2359
+ "commit_loss": 0.025955037244148854,
2360
+ "embed_loss": 0.025955037244148854,
2361
+ "codebook_utilization": 0.959716796875,
2362
+ "lr": 2.0877085445416877e-06,
2363
+ "phase": "full_vq",
2364
+ "val_recon_loss": 0.2116551651211267,
2365
+ "val_utilization": 0.94970703125,
2366
+ "time_sec": 144.72359824180603
2367
+ },
2368
+ {
2369
+ "epoch": 182,
2370
+ "loss": 0.25932408243336974,
2371
+ "recon_loss": 0.20739676288472927,
2372
+ "commit_loss": 0.025963659688507314,
2373
+ "embed_loss": 0.025963659688507314,
2374
+ "codebook_utilization": 0.959716796875,
2375
+ "lr": 1.8635966245104596e-06,
2376
+ "phase": "full_vq",
2377
+ "val_recon_loss": 0.21165913268086617,
2378
+ "val_utilization": 0.94873046875,
2379
+ "time_sec": 142.47582912445068
2380
+ },
2381
+ {
2382
+ "epoch": 183,
2383
+ "loss": 0.25927854426265734,
2384
+ "recon_loss": 0.2073975000681916,
2385
+ "commit_loss": 0.025940522172950148,
2386
+ "embed_loss": 0.025940522172950148,
2387
+ "codebook_utilization": 0.959716796875,
2388
+ "lr": 1.651978510731188e-06,
2389
+ "phase": "full_vq",
2390
+ "val_recon_loss": 0.2116663586557581,
2391
+ "val_utilization": 0.94921875,
2392
+ "time_sec": 140.14426827430725
2393
+ },
2394
+ {
2395
+ "epoch": 184,
2396
+ "loss": 0.2592831317002211,
2397
+ "recon_loss": 0.20738966548992044,
2398
+ "commit_loss": 0.02594673324648927,
2399
+ "embed_loss": 0.02594673324648927,
2400
+ "codebook_utilization": 0.959716796875,
2401
+ "lr": 1.4529091286973986e-06,
2402
+ "phase": "full_vq",
2403
+ "val_recon_loss": 0.21166383023007532,
2404
+ "val_utilization": 0.94921875,
2405
+ "time_sec": 144.8687973022461
2406
+ },
2407
+ {
2408
+ "epoch": 185,
2409
+ "loss": 0.2593069303488021,
2410
+ "recon_loss": 0.20735971904220943,
2411
+ "commit_loss": 0.025973605693678868,
2412
+ "embed_loss": 0.025973605693678868,
2413
+ "codebook_utilization": 0.959716796875,
2414
+ "lr": 1.2664401468786106e-06,
2415
+ "phase": "full_vq",
2416
+ "val_recon_loss": 0.21165352858854145,
2417
+ "val_utilization": 0.94873046875,
2418
+ "time_sec": 142.93955612182617
2419
+ },
2420
+ {
2421
+ "epoch": 186,
2422
+ "loss": 0.25926595474000225,
2423
+ "recon_loss": 0.20737890376227336,
2424
+ "commit_loss": 0.025943525233949587,
2425
+ "embed_loss": 0.025943525233949587,
2426
+ "codebook_utilization": 0.959716796875,
2427
+ "lr": 1.0926199633097149e-06,
2428
+ "phase": "full_vq",
2429
+ "val_recon_loss": 0.21167138235622576,
2430
+ "val_utilization": 0.948974609375,
2431
+ "time_sec": 139.406170129776
2432
+ },
2433
+ {
2434
+ "epoch": 187,
2435
+ "loss": 0.2593152083072882,
2436
+ "recon_loss": 0.2073678606893958,
2437
+ "commit_loss": 0.025973673816517925,
2438
+ "embed_loss": 0.025973673816517925,
2439
+ "codebook_utilization": 0.959716796875,
2440
+ "lr": 9.314936930293277e-07,
2441
+ "phase": "full_vq",
2442
+ "val_recon_loss": 0.21163264737370308,
2443
+ "val_utilization": 0.948974609375,
2444
+ "time_sec": 135.59535026550293
2445
+ },
2446
+ {
2447
+ "epoch": 188,
2448
+ "loss": 0.2592416338642761,
2449
+ "recon_loss": 0.2073135251800219,
2450
+ "commit_loss": 0.02596405419321645,
2451
+ "embed_loss": 0.02596405419321645,
2452
+ "codebook_utilization": 0.959716796875,
2453
+ "lr": 7.83103156370118e-07,
2454
+ "phase": "full_vq",
2455
+ "val_recon_loss": 0.2116496991156862,
2456
+ "val_utilization": 0.94921875,
2457
+ "time_sec": 144.72603034973145
2458
+ },
2459
+ {
2460
+ "epoch": 189,
2461
+ "loss": 0.25936319152029547,
2462
+ "recon_loss": 0.20739692338517687,
2463
+ "commit_loss": 0.02598313396407902,
2464
+ "embed_loss": 0.02598313396407902,
2465
+ "codebook_utilization": 0.959716796875,
2466
+ "lr": 6.474868681043573e-07,
2467
+ "phase": "full_vq",
2468
+ "val_recon_loss": 0.21165399198954024,
2469
+ "val_utilization": 0.94873046875,
2470
+ "time_sec": 144.5444836616516
2471
+ },
2472
+ {
2473
+ "epoch": 190,
2474
+ "loss": 0.25929971376489497,
2475
+ "recon_loss": 0.2073520045010701,
2476
+ "commit_loss": 0.025973854581434228,
2477
+ "embed_loss": 0.025973854581434228,
2478
+ "codebook_utilization": 0.959716796875,
2479
+ "lr": 5.246800274474491e-07,
2480
+ "phase": "full_vq",
2481
+ "val_recon_loss": 0.21163897752092126,
2482
+ "val_utilization": 0.94873046875,
2483
+ "time_sec": 139.40892887115479
2484
+ },
2485
+ {
2486
+ "epoch": 191,
2487
+ "loss": 0.259304106073974,
2488
+ "recon_loss": 0.20735246548807718,
2489
+ "commit_loss": 0.02597582013898991,
2490
+ "embed_loss": 0.02597582013898991,
2491
+ "codebook_utilization": 0.959716796875,
2492
+ "lr": 4.1471450892189815e-07,
2493
+ "phase": "full_vq",
2494
+ "val_recon_loss": 0.2116421643184142,
2495
+ "val_utilization": 0.948486328125,
2496
+ "time_sec": 141.0518352985382
2497
+ },
2498
+ {
2499
+ "epoch": 192,
2500
+ "loss": 0.259295791387558,
2501
+ "recon_loss": 0.20734135986262867,
2502
+ "commit_loss": 0.025977215870992762,
2503
+ "embed_loss": 0.025977215870992762,
2504
+ "codebook_utilization": 0.959716796875,
2505
+ "lr": 3.1761885408435033e-07,
2506
+ "phase": "full_vq",
2507
+ "val_recon_loss": 0.21163616413145922,
2508
+ "val_utilization": 0.949462890625,
2509
+ "time_sec": 145.0444483757019
2510
+ },
2511
+ {
2512
+ "epoch": 193,
2513
+ "loss": 0.25935043237073635,
2514
+ "recon_loss": 0.20738531138273436,
2515
+ "commit_loss": 0.025982560854920043,
2516
+ "embed_loss": 0.025982560854920043,
2517
+ "codebook_utilization": 0.959716796875,
2518
+ "lr": 2.3341826411756845e-07,
2519
+ "phase": "full_vq",
2520
+ "val_recon_loss": 0.2116364725352673,
2521
+ "val_utilization": 0.94921875,
2522
+ "time_sec": 141.71662831306458
2523
+ },
2524
+ {
2525
+ "epoch": 194,
2526
+ "loss": 0.25931752868864916,
2527
+ "recon_loss": 0.20735289103939605,
2528
+ "commit_loss": 0.025982318837246113,
2529
+ "embed_loss": 0.025982318837246113,
2530
+ "codebook_utilization": 0.959716796875,
2531
+ "lr": 1.6213459328950342e-07,
2532
+ "phase": "full_vq",
2533
+ "val_recon_loss": 0.21165755366006594,
2534
+ "val_utilization": 0.94921875,
2535
+ "time_sec": 137.2107183933258
2536
+ },
2537
+ {
2538
+ "epoch": 195,
2539
+ "loss": 0.25928719208492496,
2540
+ "recon_loss": 0.2073436098773951,
2541
+ "commit_loss": 0.025971790985141182,
2542
+ "embed_loss": 0.025971790985141182,
2543
+ "codebook_utilization": 0.959716796875,
2544
+ "lr": 1.037863432809926e-07,
2545
+ "phase": "full_vq",
2546
+ "val_recon_loss": 0.21164123923256156,
2547
+ "val_utilization": 0.948974609375,
2548
+ "time_sec": 141.30843544006348
2549
+ },
2550
+ {
2551
+ "epoch": 196,
2552
+ "loss": 0.2592993293028214,
2553
+ "recon_loss": 0.2073363337897996,
2554
+ "commit_loss": 0.025981497743891344,
2555
+ "embed_loss": 0.025981497743891344,
2556
+ "codebook_utilization": 0.959716796875,
2557
+ "lr": 5.8388658383673425e-08,
2558
+ "phase": "full_vq",
2559
+ "val_recon_loss": 0.21163905282201392,
2560
+ "val_utilization": 0.948974609375,
2561
+ "time_sec": 141.17232823371887
2562
+ },
2563
+ {
2564
+ "epoch": 197,
2565
+ "loss": 0.25932077059981623,
2566
+ "recon_loss": 0.20735098768863575,
2567
+ "commit_loss": 0.025984891571690073,
2568
+ "embed_loss": 0.025984891571690073,
2569
+ "codebook_utilization": 0.959716796875,
2570
+ "lr": 2.5953321569255322e-08,
2571
+ "phase": "full_vq",
2572
+ "val_recon_loss": 0.21163846862115218,
2573
+ "val_utilization": 0.948974609375,
2574
+ "time_sec": 142.5461483001709
2575
+ },
2576
+ {
2577
+ "epoch": 198,
2578
+ "loss": 0.25932280507271854,
2579
+ "recon_loss": 0.20735858774524393,
2580
+ "commit_loss": 0.025982108560257004,
2581
+ "embed_loss": 0.025982108560257004,
2582
+ "codebook_utilization": 0.959716796875,
2583
+ "lr": 6.4887514312661454e-09,
2584
+ "phase": "full_vq",
2585
+ "val_recon_loss": 0.21163548972834362,
2586
+ "val_utilization": 0.948974609375,
2587
+ "time_sec": 141.6607744693756
2588
+ },
2589
+ {
2590
+ "epoch": 199,
2591
+ "loss": 0.2592883176071857,
2592
+ "recon_loss": 0.207341201118822,
2593
+ "commit_loss": 0.025973558461238052,
2594
+ "embed_loss": 0.025973558461238052,
2595
+ "codebook_utilization": 0.959716796875,
2596
+ "lr": 0.0,
2597
+ "phase": "full_vq",
2598
+ "val_recon_loss": 0.21163536407304614,
2599
+ "val_utilization": 0.948974609375,
2600
+ "time_sec": 138.36240911483765
2601
+ }
2602
+ ]