Pthahnix commited on
Commit
6df68c8
·
verified ·
1 Parent(s): f5c65d2

Upload checkpoints/exp2_A_lvis_wide/training_history.json with huggingface_hub

Browse files
checkpoints/exp2_A_lvis_wide/training_history.json ADDED
@@ -0,0 +1,2602 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "epoch": 0,
4
+ "loss": 0.6107338163749312,
5
+ "recon_loss": 0.6107338163749312,
6
+ "commit_loss": 0.3489506587911104,
7
+ "embed_loss": 0.3489506587911104,
8
+ "codebook_utilization": 0.026123046875,
9
+ "lr": 2.08e-05,
10
+ "phase": "encoder_warmup",
11
+ "val_recon_loss": 0.5640084900882807,
12
+ "val_utilization": 0.016357421875,
13
+ "time_sec": 681.1810250282288
14
+ },
15
+ {
16
+ "epoch": 1,
17
+ "loss": 0.36119572697131613,
18
+ "recon_loss": 0.36119572697131613,
19
+ "commit_loss": 0.35936276113922355,
20
+ "embed_loss": 0.35936276113922355,
21
+ "codebook_utilization": 0.0625,
22
+ "lr": 4.06e-05,
23
+ "phase": "encoder_warmup",
24
+ "val_recon_loss": 0.328320093787788,
25
+ "val_utilization": 0.019775390625,
26
+ "time_sec": 678.8027560710907
27
+ },
28
+ {
29
+ "epoch": 2,
30
+ "loss": 0.3094499470176412,
31
+ "recon_loss": 0.3094499470176412,
32
+ "commit_loss": 0.34318291264659345,
33
+ "embed_loss": 0.34318291264659345,
34
+ "codebook_utilization": 0.05908203125,
35
+ "lr": 6.04e-05,
36
+ "phase": "encoder_warmup",
37
+ "val_recon_loss": 0.30003171807594514,
38
+ "val_utilization": 0.0283203125,
39
+ "time_sec": 670.3594076633453
40
+ },
41
+ {
42
+ "epoch": 3,
43
+ "loss": 0.2914164912732959,
44
+ "recon_loss": 0.2914164912732959,
45
+ "commit_loss": 0.39329903704041064,
46
+ "embed_loss": 0.39329903704041064,
47
+ "codebook_utilization": 0.048828125,
48
+ "lr": 8.02e-05,
49
+ "phase": "encoder_warmup",
50
+ "val_recon_loss": 0.28885072185082383,
51
+ "val_utilization": 0.02880859375,
52
+ "time_sec": 644.6327970027924
53
+ },
54
+ {
55
+ "epoch": 4,
56
+ "loss": 0.28421061540522224,
57
+ "recon_loss": 0.28421061540522224,
58
+ "commit_loss": 0.47273610390944854,
59
+ "embed_loss": 0.47273610390944854,
60
+ "codebook_utilization": 0.054443359375,
61
+ "lr": 0.0001,
62
+ "phase": "encoder_warmup",
63
+ "val_recon_loss": 0.2855203581994839,
64
+ "val_utilization": 0.02978515625,
65
+ "time_sec": 651.0396690368652
66
+ },
67
+ {
68
+ "epoch": 5,
69
+ "loss": 0.28003382900866064,
70
+ "recon_loss": 0.28003382900866064,
71
+ "commit_loss": 0.5642459676840764,
72
+ "embed_loss": 0.5642459676840764,
73
+ "codebook_utilization": 0.0595703125,
74
+ "lr": 9.999351124856874e-05,
75
+ "phase": "encoder_warmup",
76
+ "val_recon_loss": 0.2788614217150077,
77
+ "val_utilization": 0.043212890625,
78
+ "time_sec": 650.8557176589966
79
+ },
80
+ {
81
+ "epoch": 6,
82
+ "loss": 0.27404609167075095,
83
+ "recon_loss": 0.27404609167075095,
84
+ "commit_loss": 0.6590415956528206,
85
+ "embed_loss": 0.6590415956528206,
86
+ "codebook_utilization": 0.067138671875,
87
+ "lr": 9.997404667843075e-05,
88
+ "phase": "encoder_warmup",
89
+ "val_recon_loss": 0.2749180347564515,
90
+ "val_utilization": 0.049072265625,
91
+ "time_sec": 683.8835682868958
92
+ },
93
+ {
94
+ "epoch": 7,
95
+ "loss": 0.2685743272304535,
96
+ "recon_loss": 0.2685743272304535,
97
+ "commit_loss": 0.767206551905893,
98
+ "embed_loss": 0.767206551905893,
99
+ "codebook_utilization": 0.069091796875,
100
+ "lr": 9.994161134161632e-05,
101
+ "phase": "encoder_warmup",
102
+ "val_recon_loss": 0.27136335538679296,
103
+ "val_utilization": 0.048828125,
104
+ "time_sec": 684.5951306819916
105
+ },
106
+ {
107
+ "epoch": 8,
108
+ "loss": 0.2661038599201657,
109
+ "recon_loss": 0.2661038599201657,
110
+ "commit_loss": 0.8975046369118419,
111
+ "embed_loss": 0.8975046369118419,
112
+ "codebook_utilization": 0.069580078125,
113
+ "lr": 9.9896213656719e-05,
114
+ "phase": "encoder_warmup",
115
+ "val_recon_loss": 0.2678525708197208,
116
+ "val_utilization": 0.053466796875,
117
+ "time_sec": 692.2218220233917
118
+ },
119
+ {
120
+ "epoch": 9,
121
+ "loss": 0.2645471014466066,
122
+ "recon_loss": 0.2645471014466066,
123
+ "commit_loss": 1.0287082829288028,
124
+ "embed_loss": 1.0287082829288028,
125
+ "codebook_utilization": 0.071044921875,
126
+ "lr": 9.98378654067105e-05,
127
+ "phase": "encoder_warmup",
128
+ "val_recon_loss": 0.6881499963529994,
129
+ "val_utilization": 0.95458984375,
130
+ "time_sec": 699.1975193023682
131
+ },
132
+ {
133
+ "epoch": 10,
134
+ "loss": 0.3321987629582888,
135
+ "recon_loss": 0.2864969655629096,
136
+ "commit_loss": 0.022850898831456818,
137
+ "embed_loss": 0.022850898831456818,
138
+ "codebook_utilization": 0.965576171875,
139
+ "lr": 9.976658173588243e-05,
140
+ "phase": "full_vq",
141
+ "val_recon_loss": 0.2551815191681465,
142
+ "val_utilization": 0.731201171875,
143
+ "time_sec": 680.008100271225
144
+ },
145
+ {
146
+ "epoch": 11,
147
+ "loss": 0.27075304393845845,
148
+ "recon_loss": 0.2473720870609206,
149
+ "commit_loss": 0.011690477925153363,
150
+ "embed_loss": 0.011690477925153363,
151
+ "codebook_utilization": 0.900390625,
152
+ "lr": 9.968238114591565e-05,
153
+ "phase": "full_vq",
154
+ "val_recon_loss": 0.2458065699660376,
155
+ "val_utilization": 0.82177734375,
156
+ "time_sec": 683.6505243778229
157
+ },
158
+ {
159
+ "epoch": 12,
160
+ "loss": 0.2626437447904571,
161
+ "recon_loss": 0.24087850822181237,
162
+ "commit_loss": 0.01088261839916026,
163
+ "embed_loss": 0.01088261839916026,
164
+ "codebook_utilization": 0.9306640625,
165
+ "lr": 9.95852854910781e-05,
166
+ "phase": "full_vq",
167
+ "val_recon_loss": 0.24387330520019102,
168
+ "val_utilization": 0.851318359375,
169
+ "time_sec": 672.1745018959045
170
+ },
171
+ {
172
+ "epoch": 13,
173
+ "loss": 0.25929648746158374,
174
+ "recon_loss": 0.2375372385348731,
175
+ "commit_loss": 0.01087962448354659,
176
+ "embed_loss": 0.01087962448354659,
177
+ "codebook_utilization": 0.944091796875,
178
+ "lr": 9.947531997255256e-05,
179
+ "phase": "full_vq",
180
+ "val_recon_loss": 0.23887540533971252,
181
+ "val_utilization": 0.889892578125,
182
+ "time_sec": 674.5686082839966
183
+ },
184
+ {
185
+ "epoch": 14,
186
+ "loss": 0.2575729920612118,
187
+ "recon_loss": 0.2352601911438513,
188
+ "commit_loss": 0.01115640010533294,
189
+ "embed_loss": 0.01115640010533294,
190
+ "codebook_utilization": 0.95166015625,
191
+ "lr": 9.935251313189564e-05,
192
+ "phase": "full_vq",
193
+ "val_recon_loss": 0.23747419591030378,
194
+ "val_utilization": 0.890625,
195
+ "time_sec": 682.1447324752808
196
+ },
197
+ {
198
+ "epoch": 15,
199
+ "loss": 0.2567580191262046,
200
+ "recon_loss": 0.23352349991720867,
201
+ "commit_loss": 0.011617259449277547,
202
+ "embed_loss": 0.011617259449277547,
203
+ "codebook_utilization": 0.95556640625,
204
+ "lr": 9.921689684362989e-05,
205
+ "phase": "full_vq",
206
+ "val_recon_loss": 0.23640040006865276,
207
+ "val_utilization": 0.909912109375,
208
+ "time_sec": 679.4425365924835
209
+ },
210
+ {
211
+ "epoch": 16,
212
+ "loss": 0.2560225958787006,
213
+ "recon_loss": 0.2319405957612242,
214
+ "commit_loss": 0.012041000058738196,
215
+ "embed_loss": 0.012041000058738196,
216
+ "codebook_utilization": 0.95849609375,
217
+ "lr": 9.906850630697068e-05,
218
+ "phase": "full_vq",
219
+ "val_recon_loss": 0.23416379002038012,
220
+ "val_utilization": 0.92041015625,
221
+ "time_sec": 669.8222999572754
222
+ },
223
+ {
224
+ "epoch": 17,
225
+ "loss": 0.25528010808759266,
226
+ "recon_loss": 0.23040353310948142,
227
+ "commit_loss": 0.012438287347716692,
228
+ "embed_loss": 0.012438287347716692,
229
+ "codebook_utilization": 0.964111328125,
230
+ "lr": 9.890738003669029e-05,
231
+ "phase": "full_vq",
232
+ "val_recon_loss": 0.2320063937915845,
233
+ "val_utilization": 0.923095703125,
234
+ "time_sec": 687.6758215427399
235
+ },
236
+ {
237
+ "epoch": 18,
238
+ "loss": 0.2550786417230035,
239
+ "recon_loss": 0.22934870264394497,
240
+ "commit_loss": 0.012864969803277797,
241
+ "embed_loss": 0.012864969803277797,
242
+ "codebook_utilization": 0.96728515625,
243
+ "lr": 9.87335598531214e-05,
244
+ "phase": "full_vq",
245
+ "val_recon_loss": 0.23130840858381785,
246
+ "val_utilization": 0.92578125,
247
+ "time_sec": 702.5455520153046
248
+ },
249
+ {
250
+ "epoch": 19,
251
+ "loss": 0.25469206945925227,
252
+ "recon_loss": 0.22842017553120772,
253
+ "commit_loss": 0.013135946958974448,
254
+ "embed_loss": 0.013135946958974448,
255
+ "codebook_utilization": 0.96826171875,
256
+ "lr": 9.85470908713026e-05,
257
+ "phase": "full_vq",
258
+ "val_recon_loss": 0.23006552106209016,
259
+ "val_utilization": 0.933349609375,
260
+ "time_sec": 694.8513193130493
261
+ },
262
+ {
263
+ "epoch": 20,
264
+ "loss": 0.25481168806633653,
265
+ "recon_loss": 0.22772028184114756,
266
+ "commit_loss": 0.013545703335960464,
267
+ "embed_loss": 0.013545703335960464,
268
+ "codebook_utilization": 0.96484375,
269
+ "lr": 9.834802148926882e-05,
270
+ "phase": "full_vq",
271
+ "val_recon_loss": 0.23030991994597938,
272
+ "val_utilization": 0.94580078125,
273
+ "time_sec": 135.60633659362793
274
+ },
275
+ {
276
+ "epoch": 21,
277
+ "loss": 0.2549404268705748,
278
+ "recon_loss": 0.22707932125423658,
279
+ "commit_loss": 0.0139305527804061,
280
+ "embed_loss": 0.0139305527804061,
281
+ "codebook_utilization": 0.9658203125,
282
+ "lr": 9.813640337548953e-05,
283
+ "phase": "full_vq",
284
+ "val_recon_loss": 0.22884476528073963,
285
+ "val_utilization": 0.935546875,
286
+ "time_sec": 133.3524444103241
287
+ },
288
+ {
289
+ "epoch": 22,
290
+ "loss": 0.2549613991646263,
291
+ "recon_loss": 0.2261909286019602,
292
+ "commit_loss": 0.014385235286380863,
293
+ "embed_loss": 0.014385235286380863,
294
+ "codebook_utilization": 0.9658203125,
295
+ "lr": 9.79122914554583e-05,
296
+ "phase": "full_vq",
297
+ "val_recon_loss": 0.2293451830242457,
298
+ "val_utilization": 0.934814453125,
299
+ "time_sec": 142.04170513153076
300
+ },
301
+ {
302
+ "epoch": 23,
303
+ "loss": 0.255592231989553,
304
+ "recon_loss": 0.22587027725811573,
305
+ "commit_loss": 0.01486097723068947,
306
+ "embed_loss": 0.01486097723068947,
307
+ "codebook_utilization": 0.96630859375,
308
+ "lr": 9.76757438974368e-05,
309
+ "phase": "full_vq",
310
+ "val_recon_loss": 0.22918162160040287,
311
+ "val_utilization": 0.9423828125,
312
+ "time_sec": 143.15011739730835
313
+ },
314
+ {
315
+ "epoch": 24,
316
+ "loss": 0.25588738506402425,
317
+ "recon_loss": 0.22522058667044653,
318
+ "commit_loss": 0.015333399037782577,
319
+ "embed_loss": 0.015333399037782577,
320
+ "codebook_utilization": 0.966064453125,
321
+ "lr": 9.742682209735726e-05,
322
+ "phase": "full_vq",
323
+ "val_recon_loss": 0.22730647220035619,
324
+ "val_utilization": 0.94482421875,
325
+ "time_sec": 133.78786826133728
326
+ },
327
+ {
328
+ "epoch": 25,
329
+ "loss": 0.25651584123450566,
330
+ "recon_loss": 0.22474570039326583,
331
+ "commit_loss": 0.01588507030325813,
332
+ "embed_loss": 0.01588507030325813,
333
+ "codebook_utilization": 0.966064453125,
334
+ "lr": 9.716559066288714e-05,
335
+ "phase": "full_vq",
336
+ "val_recon_loss": 0.22761344323667249,
337
+ "val_utilization": 0.940673828125,
338
+ "time_sec": 145.52776169776917
339
+ },
340
+ {
341
+ "epoch": 26,
342
+ "loss": 0.25648680357590603,
343
+ "recon_loss": 0.22416417682316245,
344
+ "commit_loss": 0.016161313402872825,
345
+ "embed_loss": 0.016161313402872825,
346
+ "codebook_utilization": 0.9658203125,
347
+ "lr": 9.689211739666022e-05,
348
+ "phase": "full_vq",
349
+ "val_recon_loss": 0.22609426340695177,
350
+ "val_utilization": 0.950927734375,
351
+ "time_sec": 144.49739360809326
352
+ },
353
+ {
354
+ "epoch": 27,
355
+ "loss": 0.25732946254535094,
356
+ "recon_loss": 0.22391341420693125,
357
+ "commit_loss": 0.016708024692921052,
358
+ "embed_loss": 0.016708024692921052,
359
+ "codebook_utilization": 0.9658203125,
360
+ "lr": 9.660647327867839e-05,
361
+ "phase": "full_vq",
362
+ "val_recon_loss": 0.2264180713322725,
363
+ "val_utilization": 0.94287109375,
364
+ "time_sec": 137.48128390312195
365
+ },
366
+ {
367
+ "epoch": 28,
368
+ "loss": 0.25810114415438196,
369
+ "recon_loss": 0.22355460260457138,
370
+ "commit_loss": 0.017273270694140173,
371
+ "embed_loss": 0.017273270694140173,
372
+ "codebook_utilization": 0.966552734375,
373
+ "lr": 9.630873244788881e-05,
374
+ "phase": "full_vq",
375
+ "val_recon_loss": 0.22547388160496615,
376
+ "val_utilization": 0.947509765625,
377
+ "time_sec": 148.51082181930542
378
+ },
379
+ {
380
+ "epoch": 29,
381
+ "loss": 0.25829041187921514,
382
+ "recon_loss": 0.22303170099006436,
383
+ "commit_loss": 0.017629355568246952,
384
+ "embed_loss": 0.017629355568246952,
385
+ "codebook_utilization": 0.965576171875,
386
+ "lr": 9.59989721829412e-05,
387
+ "phase": "full_vq",
388
+ "val_recon_loss": 0.22596221567874544,
389
+ "val_utilization": 0.949462890625,
390
+ "time_sec": 139.81672883033752
391
+ },
392
+ {
393
+ "epoch": 30,
394
+ "loss": 0.2585969795056475,
395
+ "recon_loss": 0.2226741507972482,
396
+ "commit_loss": 0.017961414149762978,
397
+ "embed_loss": 0.017961414149762978,
398
+ "codebook_utilization": 0.965087890625,
399
+ "lr": 9.567727288213003e-05,
400
+ "phase": "full_vq",
401
+ "val_recon_loss": 0.22464002208428435,
402
+ "val_utilization": 0.941650390625,
403
+ "time_sec": 144.46557545661926
404
+ },
405
+ {
406
+ "epoch": 31,
407
+ "loss": 0.25888408274266134,
408
+ "recon_loss": 0.2221761466405256,
409
+ "commit_loss": 0.018353968197454607,
410
+ "embed_loss": 0.018353968197454607,
411
+ "codebook_utilization": 0.96533203125,
412
+ "lr": 9.534371804252726e-05,
413
+ "phase": "full_vq",
414
+ "val_recon_loss": 0.22492716059590992,
415
+ "val_utilization": 0.94482421875,
416
+ "time_sec": 140.9813690185547
417
+ },
418
+ {
419
+ "epoch": 32,
420
+ "loss": 0.2596231485568088,
421
+ "recon_loss": 0.2221419390664514,
422
+ "commit_loss": 0.018740604543265933,
423
+ "embed_loss": 0.018740604543265933,
424
+ "codebook_utilization": 0.965087890625,
425
+ "lr": 9.499839423831059e-05,
426
+ "phase": "full_vq",
427
+ "val_recon_loss": 0.22403986592975894,
428
+ "val_utilization": 0.94873046875,
429
+ "time_sec": 141.7025327682495
430
+ },
431
+ {
432
+ "epoch": 33,
433
+ "loss": 0.25944470799954605,
434
+ "recon_loss": 0.2217554383000061,
435
+ "commit_loss": 0.018844635066826168,
436
+ "embed_loss": 0.018844635066826168,
437
+ "codebook_utilization": 0.965576171875,
438
+ "lr": 9.464139109829319e-05,
439
+ "phase": "full_vq",
440
+ "val_recon_loss": 0.2241385518834832,
441
+ "val_utilization": 0.945556640625,
442
+ "time_sec": 142.47765183448792
443
+ },
444
+ {
445
+ "epoch": 34,
446
+ "loss": 0.25979276433546694,
447
+ "recon_loss": 0.22145432936466805,
448
+ "commit_loss": 0.019169217614118807,
449
+ "embed_loss": 0.019169217614118807,
450
+ "codebook_utilization": 0.965576171875,
451
+ "lr": 9.427280128266047e-05,
452
+ "phase": "full_vq",
453
+ "val_recon_loss": 0.22376884955368684,
454
+ "val_utilization": 0.94775390625,
455
+ "time_sec": 139.88372087478638
456
+ },
457
+ {
458
+ "epoch": 35,
459
+ "loss": 0.2599717076554854,
460
+ "recon_loss": 0.22109364320430652,
461
+ "commit_loss": 0.019439032563793302,
462
+ "embed_loss": 0.019439032563793302,
463
+ "codebook_utilization": 0.965576171875,
464
+ "lr": 9.389272045892021e-05,
465
+ "phase": "full_vq",
466
+ "val_recon_loss": 0.22392536733257637,
467
+ "val_utilization": 0.949462890625,
468
+ "time_sec": 140.3261411190033
469
+ },
470
+ {
471
+ "epoch": 36,
472
+ "loss": 0.26018684066650344,
473
+ "recon_loss": 0.22100506206030446,
474
+ "commit_loss": 0.01959088937124505,
475
+ "embed_loss": 0.01959088937124505,
476
+ "codebook_utilization": 0.96533203125,
477
+ "lr": 9.350124727707194e-05,
478
+ "phase": "full_vq",
479
+ "val_recon_loss": 0.22288394819819526,
480
+ "val_utilization": 0.9482421875,
481
+ "time_sec": 140.9074785709381
482
+ },
483
+ {
484
+ "epoch": 37,
485
+ "loss": 0.26017498877151873,
486
+ "recon_loss": 0.22058909200636675,
487
+ "commit_loss": 0.019792948420434626,
488
+ "embed_loss": 0.019792948420434626,
489
+ "codebook_utilization": 0.96533203125,
490
+ "lr": 9.309848334400244e-05,
491
+ "phase": "full_vq",
492
+ "val_recon_loss": 0.22249055863096473,
493
+ "val_utilization": 0.952880859375,
494
+ "time_sec": 142.68842220306396
495
+ },
496
+ {
497
+ "epoch": 38,
498
+ "loss": 0.26064721414148,
499
+ "recon_loss": 0.22056346608775096,
500
+ "commit_loss": 0.02004187426411201,
501
+ "embed_loss": 0.02004187426411201,
502
+ "codebook_utilization": 0.96533203125,
503
+ "lr": 9.26845331971136e-05,
504
+ "phase": "full_vq",
505
+ "val_recon_loss": 0.22398295328858192,
506
+ "val_utilization": 0.951416015625,
507
+ "time_sec": 143.05380296707153
508
+ },
509
+ {
510
+ "epoch": 39,
511
+ "loss": 0.260869705479158,
512
+ "recon_loss": 0.22040401851420158,
513
+ "commit_loss": 0.020232843641484494,
514
+ "embed_loss": 0.020232843641484494,
515
+ "codebook_utilization": 0.96533203125,
516
+ "lr": 9.225950427718973e-05,
517
+ "phase": "full_vq",
518
+ "val_recon_loss": 0.22268290390794196,
519
+ "val_utilization": 0.947265625,
520
+ "time_sec": 137.99860167503357
521
+ },
522
+ {
523
+ "epoch": 40,
524
+ "loss": 0.2607682509635522,
525
+ "recon_loss": 0.21996311734362348,
526
+ "commit_loss": 0.020402566807440466,
527
+ "embed_loss": 0.020402566807440466,
528
+ "codebook_utilization": 0.965087890625,
529
+ "lr": 9.182350690051132e-05,
530
+ "phase": "full_vq",
531
+ "val_recon_loss": 0.22248972224050692,
532
+ "val_utilization": 0.947998046875,
533
+ "time_sec": 143.34028887748718
534
+ },
535
+ {
536
+ "epoch": 41,
537
+ "loss": 0.2611755917711956,
538
+ "recon_loss": 0.21986291555531304,
539
+ "commit_loss": 0.020656338024652263,
540
+ "embed_loss": 0.020656338024652263,
541
+ "codebook_utilization": 0.96533203125,
542
+ "lr": 9.137665423022247e-05,
543
+ "phase": "full_vq",
544
+ "val_recon_loss": 0.22248069875026016,
545
+ "val_utilization": 0.95361328125,
546
+ "time_sec": 142.6788067817688
547
+ },
548
+ {
549
+ "epoch": 42,
550
+ "loss": 0.2613665052464984,
551
+ "recon_loss": 0.21954387908947823,
552
+ "commit_loss": 0.0209113132703272,
553
+ "embed_loss": 0.0209113132703272,
554
+ "codebook_utilization": 0.965087890625,
555
+ "lr": 9.091906224695934e-05,
556
+ "phase": "full_vq",
557
+ "val_recon_loss": 0.22234819837835398,
558
+ "val_utilization": 0.9541015625,
559
+ "time_sec": 143.6315414905548
560
+ },
561
+ {
562
+ "epoch": 43,
563
+ "loss": 0.26161319392968,
564
+ "recon_loss": 0.2195290318591808,
565
+ "commit_loss": 0.02104208134316655,
566
+ "embed_loss": 0.02104208134316655,
567
+ "codebook_utilization": 0.965087890625,
568
+ "lr": 9.045084971874735e-05,
569
+ "phase": "full_vq",
570
+ "val_recon_loss": 0.22122458452254198,
571
+ "val_utilization": 0.95166015625,
572
+ "time_sec": 140.05932068824768
573
+ },
574
+ {
575
+ "epoch": 44,
576
+ "loss": 0.26158433698299455,
577
+ "recon_loss": 0.21912217705553463,
578
+ "commit_loss": 0.021231079958682137,
579
+ "embed_loss": 0.021231079958682137,
580
+ "codebook_utilization": 0.96533203125,
581
+ "lr": 8.997213817017505e-05,
582
+ "phase": "full_vq",
583
+ "val_recon_loss": 0.22099923769409738,
584
+ "val_utilization": 0.952392578125,
585
+ "time_sec": 139.51053547859192
586
+ },
587
+ {
588
+ "epoch": 45,
589
+ "loss": 0.26194193187929427,
590
+ "recon_loss": 0.21911258894576613,
591
+ "commit_loss": 0.02141467151471835,
592
+ "embed_loss": 0.02141467151471835,
593
+ "codebook_utilization": 0.96533203125,
594
+ "lr": 8.948305185085224e-05,
595
+ "phase": "full_vq",
596
+ "val_recon_loss": 0.22119122241320235,
597
+ "val_utilization": 0.953125,
598
+ "time_sec": 134.75310683250427
599
+ },
600
+ {
601
+ "epoch": 46,
602
+ "loss": 0.2620307810019025,
603
+ "recon_loss": 0.2188515700500832,
604
+ "commit_loss": 0.021589605612200774,
605
+ "embed_loss": 0.021589605612200774,
606
+ "codebook_utilization": 0.96533203125,
607
+ "lr": 8.89837177031611e-05,
608
+ "phase": "full_vq",
609
+ "val_recon_loss": 0.22111950647295192,
610
+ "val_utilization": 0.948486328125,
611
+ "time_sec": 139.76739192008972
612
+ },
613
+ {
614
+ "epoch": 47,
615
+ "loss": 0.2620187394704599,
616
+ "recon_loss": 0.21868553892949086,
617
+ "commit_loss": 0.021666600209910694,
618
+ "embed_loss": 0.021666600209910694,
619
+ "codebook_utilization": 0.965087890625,
620
+ "lr": 8.847426532930828e-05,
621
+ "phase": "full_vq",
622
+ "val_recon_loss": 0.2209378086616484,
623
+ "val_utilization": 0.955322265625,
624
+ "time_sec": 137.01194143295288
625
+ },
626
+ {
627
+ "epoch": 48,
628
+ "loss": 0.2621924214892917,
629
+ "recon_loss": 0.21844577922569058,
630
+ "commit_loss": 0.021873321210041763,
631
+ "embed_loss": 0.021873321210041763,
632
+ "codebook_utilization": 0.965087890625,
633
+ "lr": 8.795482695768655e-05,
634
+ "phase": "full_vq",
635
+ "val_recon_loss": 0.2208881014853381,
636
+ "val_utilization": 0.949951171875,
637
+ "time_sec": 141.00962376594543
638
+ },
639
+ {
640
+ "epoch": 49,
641
+ "loss": 0.26215042862384946,
642
+ "recon_loss": 0.21819759291523516,
643
+ "commit_loss": 0.021976417889641876,
644
+ "embed_loss": 0.021976417889641876,
645
+ "codebook_utilization": 0.965087890625,
646
+ "lr": 8.742553740855503e-05,
647
+ "phase": "full_vq",
648
+ "val_recon_loss": 0.22014236868767256,
649
+ "val_utilization": 0.952392578125,
650
+ "time_sec": 135.42713904380798
651
+ },
652
+ {
653
+ "epoch": 50,
654
+ "loss": 0.2623459170181254,
655
+ "recon_loss": 0.21805116693744167,
656
+ "commit_loss": 0.02214737502015057,
657
+ "embed_loss": 0.02214737502015057,
658
+ "codebook_utilization": 0.965087890625,
659
+ "lr": 8.688653405904649e-05,
660
+ "phase": "full_vq",
661
+ "val_recon_loss": 0.2203257353453154,
662
+ "val_utilization": 0.949951171875,
663
+ "time_sec": 140.37291860580444
664
+ },
665
+ {
666
+ "epoch": 51,
667
+ "loss": 0.26253410511546665,
668
+ "recon_loss": 0.21801359600167933,
669
+ "commit_loss": 0.022260254597276208,
670
+ "embed_loss": 0.022260254597276208,
671
+ "codebook_utilization": 0.96484375,
672
+ "lr": 8.633795680751113e-05,
673
+ "phase": "full_vq",
674
+ "val_recon_loss": 0.22066639204708377,
675
+ "val_utilization": 0.95068359375,
676
+ "time_sec": 134.7635838985443
677
+ },
678
+ {
679
+ "epoch": 52,
680
+ "loss": 0.26251633525863893,
681
+ "recon_loss": 0.2178837649907846,
682
+ "commit_loss": 0.022316285335839927,
683
+ "embed_loss": 0.022316285335839927,
684
+ "codebook_utilization": 0.96484375,
685
+ "lr": 8.577994803720604e-05,
686
+ "phase": "full_vq",
687
+ "val_recon_loss": 0.2205105960536539,
688
+ "val_utilization": 0.95068359375,
689
+ "time_sec": 132.1221911907196
690
+ },
691
+ {
692
+ "epoch": 53,
693
+ "loss": 0.2626252629488787,
694
+ "recon_loss": 0.21770087371108332,
695
+ "commit_loss": 0.02246219483595392,
696
+ "embed_loss": 0.02246219483595392,
697
+ "codebook_utilization": 0.965087890625,
698
+ "lr": 8.521265257933945e-05,
699
+ "phase": "full_vq",
700
+ "val_recon_loss": 0.22047662584299452,
701
+ "val_utilization": 0.94482421875,
702
+ "time_sec": 135.10088562965393
703
+ },
704
+ {
705
+ "epoch": 54,
706
+ "loss": 0.262684616777632,
707
+ "recon_loss": 0.2176037282357371,
708
+ "commit_loss": 0.02254044430628217,
709
+ "embed_loss": 0.02254044430628217,
710
+ "codebook_utilization": 0.96484375,
711
+ "lr": 8.463621767547996e-05,
712
+ "phase": "full_vq",
713
+ "val_recon_loss": 0.22083497809225253,
714
+ "val_utilization": 0.94775390625,
715
+ "time_sec": 134.99020910263062
716
+ },
717
+ {
718
+ "epoch": 55,
719
+ "loss": 0.2625497348788308,
720
+ "recon_loss": 0.2174131958909474,
721
+ "commit_loss": 0.022568269567135067,
722
+ "embed_loss": 0.022568269567135067,
723
+ "codebook_utilization": 0.965087890625,
724
+ "lr": 8.405079293933984e-05,
725
+ "phase": "full_vq",
726
+ "val_recon_loss": 0.21972426370288548,
727
+ "val_utilization": 0.952392578125,
728
+ "time_sec": 142.66574263572693
729
+ },
730
+ {
731
+ "epoch": 56,
732
+ "loss": 0.2625630740227738,
733
+ "recon_loss": 0.21731568442854454,
734
+ "commit_loss": 0.022623695029314293,
735
+ "embed_loss": 0.022623695029314293,
736
+ "codebook_utilization": 0.96484375,
737
+ "lr": 8.345653031794289e-05,
738
+ "phase": "full_vq",
739
+ "val_recon_loss": 0.21945370809080894,
740
+ "val_utilization": 0.953125,
741
+ "time_sec": 145.99305129051208
742
+ },
743
+ {
744
+ "epoch": 57,
745
+ "loss": 0.26258070134938893,
746
+ "recon_loss": 0.21708292392938117,
747
+ "commit_loss": 0.02274888864185833,
748
+ "embed_loss": 0.02274888864185833,
749
+ "codebook_utilization": 0.96484375,
750
+ "lr": 8.285358405218651e-05,
751
+ "phase": "full_vq",
752
+ "val_recon_loss": 0.21932257418887,
753
+ "val_utilization": 0.95556640625,
754
+ "time_sec": 138.1978497505188
755
+ },
756
+ {
757
+ "epoch": 58,
758
+ "loss": 0.2626232503633189,
759
+ "recon_loss": 0.2169914725753996,
760
+ "commit_loss": 0.022815888944437833,
761
+ "embed_loss": 0.022815888944437833,
762
+ "codebook_utilization": 0.96484375,
763
+ "lr": 8.22421106368085e-05,
764
+ "phase": "full_vq",
765
+ "val_recon_loss": 0.21894806288601307,
766
+ "val_utilization": 0.951416015625,
767
+ "time_sec": 145.86574983596802
768
+ },
769
+ {
770
+ "epoch": 59,
771
+ "loss": 0.2626482349060738,
772
+ "recon_loss": 0.2167674576848503,
773
+ "commit_loss": 0.022940388597992217,
774
+ "embed_loss": 0.022940388597992217,
775
+ "codebook_utilization": 0.96484375,
776
+ "lr": 8.162226877976883e-05,
777
+ "phase": "full_vq",
778
+ "val_recon_loss": 0.21943318760127165,
779
+ "val_utilization": 0.953857421875,
780
+ "time_sec": 142.01206827163696
781
+ },
782
+ {
783
+ "epoch": 60,
784
+ "loss": 0.26275727599051585,
785
+ "recon_loss": 0.21678639815105655,
786
+ "commit_loss": 0.022985439174644507,
787
+ "embed_loss": 0.022985439174644507,
788
+ "codebook_utilization": 0.964599609375,
789
+ "lr": 8.099421936105698e-05,
790
+ "phase": "full_vq",
791
+ "val_recon_loss": 0.21886526299326609,
792
+ "val_utilization": 0.952880859375,
793
+ "time_sec": 132.8587257862091
794
+ },
795
+ {
796
+ "epoch": 61,
797
+ "loss": 0.2626901668544384,
798
+ "recon_loss": 0.21660732560687596,
799
+ "commit_loss": 0.023041420631352963,
800
+ "embed_loss": 0.023041420631352963,
801
+ "codebook_utilization": 0.96484375,
802
+ "lr": 8.035812539093553e-05,
803
+ "phase": "full_vq",
804
+ "val_recon_loss": 0.2193846743595734,
805
+ "val_utilization": 0.947509765625,
806
+ "time_sec": 134.97693610191345
807
+ },
808
+ {
809
+ "epoch": 62,
810
+ "loss": 0.26267470548631056,
811
+ "recon_loss": 0.21658330828678318,
812
+ "commit_loss": 0.023045698645194047,
813
+ "embed_loss": 0.023045698645194047,
814
+ "codebook_utilization": 0.964599609375,
815
+ "lr": 7.971415196763084e-05,
816
+ "phase": "full_vq",
817
+ "val_recon_loss": 0.2193064247624258,
818
+ "val_utilization": 0.953125,
819
+ "time_sec": 133.91056394577026
820
+ },
821
+ {
822
+ "epoch": 63,
823
+ "loss": 0.2625721351806387,
824
+ "recon_loss": 0.21640123990250798,
825
+ "commit_loss": 0.023085447757689125,
826
+ "embed_loss": 0.023085447757689125,
827
+ "codebook_utilization": 0.964111328125,
828
+ "lr": 7.90624662344818e-05,
829
+ "phase": "full_vq",
830
+ "val_recon_loss": 0.21904683314012677,
831
+ "val_utilization": 0.9541015625,
832
+ "time_sec": 135.78576517105103
833
+ },
834
+ {
835
+ "epoch": 64,
836
+ "loss": 0.26236986173039206,
837
+ "recon_loss": 0.2162628132839836,
838
+ "commit_loss": 0.023053524303969328,
839
+ "embed_loss": 0.023053524303969328,
840
+ "codebook_utilization": 0.96435546875,
841
+ "lr": 7.840323733655777e-05,
842
+ "phase": "full_vq",
843
+ "val_recon_loss": 0.21925017639492336,
844
+ "val_utilization": 0.943359375,
845
+ "time_sec": 138.8038284778595
846
+ },
847
+ {
848
+ "epoch": 65,
849
+ "loss": 0.26251463022658494,
850
+ "recon_loss": 0.216217305460758,
851
+ "commit_loss": 0.023148662216335453,
852
+ "embed_loss": 0.023148662216335453,
853
+ "codebook_utilization": 0.96435546875,
854
+ "lr": 7.773663637675692e-05,
855
+ "phase": "full_vq",
856
+ "val_recon_loss": 0.21831443294715347,
857
+ "val_utilization": 0.95263671875,
858
+ "time_sec": 134.38734650611877
859
+ },
860
+ {
861
+ "epoch": 66,
862
+ "loss": 0.26254441928330474,
863
+ "recon_loss": 0.21614270881424105,
864
+ "commit_loss": 0.023200855221912304,
865
+ "embed_loss": 0.023200855221912304,
866
+ "codebook_utilization": 0.964599609375,
867
+ "lr": 7.706283637139655e-05,
868
+ "phase": "full_vq",
869
+ "val_recon_loss": 0.21837528234117487,
870
+ "val_utilization": 0.9501953125,
871
+ "time_sec": 136.06812143325806
872
+ },
873
+ {
874
+ "epoch": 67,
875
+ "loss": 0.26247545271142714,
876
+ "recon_loss": 0.21597563437006984,
877
+ "commit_loss": 0.02324990932716102,
878
+ "embed_loss": 0.02324990932716102,
879
+ "codebook_utilization": 0.96435546875,
880
+ "lr": 7.638201220530662e-05,
881
+ "phase": "full_vq",
882
+ "val_recon_loss": 0.2185103822959943,
883
+ "val_utilization": 0.952392578125,
884
+ "time_sec": 142.83888840675354
885
+ },
886
+ {
887
+ "epoch": 68,
888
+ "loss": 0.26241777909965047,
889
+ "recon_loss": 0.2158464916999424,
890
+ "commit_loss": 0.02328564363928022,
891
+ "embed_loss": 0.02328564363928022,
892
+ "codebook_utilization": 0.96484375,
893
+ "lr": 7.569434058643842e-05,
894
+ "phase": "full_vq",
895
+ "val_recon_loss": 0.2178707250885749,
896
+ "val_utilization": 0.94775390625,
897
+ "time_sec": 137.47662663459778
898
+ },
899
+ {
900
+ "epoch": 69,
901
+ "loss": 0.26233261960954846,
902
+ "recon_loss": 0.21578528340350644,
903
+ "commit_loss": 0.023273668214073027,
904
+ "embed_loss": 0.023273668214073027,
905
+ "codebook_utilization": 0.964599609375,
906
+ "lr": 7.499999999999998e-05,
907
+ "phase": "full_vq",
908
+ "val_recon_loss": 0.21815377281288081,
909
+ "val_utilization": 0.95166015625,
910
+ "time_sec": 137.08377933502197
911
+ },
912
+ {
913
+ "epoch": 70,
914
+ "loss": 0.26238782917904013,
915
+ "recon_loss": 0.21565110981464386,
916
+ "commit_loss": 0.023368359535811394,
917
+ "embed_loss": 0.023368359535811394,
918
+ "codebook_utilization": 0.964111328125,
919
+ "lr": 7.429917066213028e-05,
920
+ "phase": "full_vq",
921
+ "val_recon_loss": 0.2190959115376633,
922
+ "val_utilization": 0.953369140625,
923
+ "time_sec": 136.61912369728088
924
+ },
925
+ {
926
+ "epoch": 71,
927
+ "loss": 0.26244070697929156,
928
+ "recon_loss": 0.21563088160947086,
929
+ "commit_loss": 0.023404912679862524,
930
+ "embed_loss": 0.023404912679862524,
931
+ "codebook_utilization": 0.96435546875,
932
+ "lr": 7.359203447312409e-05,
933
+ "phase": "full_vq",
934
+ "val_recon_loss": 0.21785652670967445,
935
+ "val_utilization": 0.9541015625,
936
+ "time_sec": 143.15168595314026
937
+ },
938
+ {
939
+ "epoch": 72,
940
+ "loss": 0.26219113080239875,
941
+ "recon_loss": 0.21544426015318247,
942
+ "commit_loss": 0.0233734353649907,
943
+ "embed_loss": 0.0233734353649907,
944
+ "codebook_utilization": 0.964599609375,
945
+ "lr": 7.287877497021975e-05,
946
+ "phase": "full_vq",
947
+ "val_recon_loss": 0.21749303205294557,
948
+ "val_utilization": 0.953857421875,
949
+ "time_sec": 132.9413378238678
950
+ },
951
+ {
952
+ "epoch": 73,
953
+ "loss": 0.2623570311642921,
954
+ "recon_loss": 0.21531846103710212,
955
+ "commit_loss": 0.023519285093881896,
956
+ "embed_loss": 0.023519285093881896,
957
+ "codebook_utilization": 0.96435546875,
958
+ "lr": 7.215957727996207e-05,
959
+ "phase": "full_vq",
960
+ "val_recon_loss": 0.21846277727169938,
961
+ "val_utilization": 0.949462890625,
962
+ "time_sec": 141.15300703048706
963
+ },
964
+ {
965
+ "epoch": 74,
966
+ "loss": 0.2624412690034404,
967
+ "recon_loss": 0.21534772207178074,
968
+ "commit_loss": 0.023546773286632246,
969
+ "embed_loss": 0.023546773286632246,
970
+ "codebook_utilization": 0.96435546875,
971
+ "lr": 7.143462807015271e-05,
972
+ "phase": "full_vq",
973
+ "val_recon_loss": 0.21801655027973518,
974
+ "val_utilization": 0.953369140625,
975
+ "time_sec": 144.36809396743774
976
+ },
977
+ {
978
+ "epoch": 75,
979
+ "loss": 0.26219636684430003,
980
+ "recon_loss": 0.2151943643685925,
981
+ "commit_loss": 0.023501001030893184,
982
+ "embed_loss": 0.023501001030893184,
983
+ "codebook_utilization": 0.964111328125,
984
+ "lr": 7.07041155014006e-05,
985
+ "phase": "full_vq",
986
+ "val_recon_loss": 0.21753000794501787,
987
+ "val_utilization": 0.9521484375,
988
+ "time_sec": 142.3067388534546
989
+ },
990
+ {
991
+ "epoch": 76,
992
+ "loss": 0.26240552162251823,
993
+ "recon_loss": 0.21510267409124995,
994
+ "commit_loss": 0.023651423874162237,
995
+ "embed_loss": 0.023651423874162237,
996
+ "codebook_utilization": 0.964111328125,
997
+ "lr": 6.996822917828478e-05,
998
+ "phase": "full_vq",
999
+ "val_recon_loss": 0.2177299749148026,
1000
+ "val_utilization": 0.95068359375,
1001
+ "time_sec": 144.21056818962097
1002
+ },
1003
+ {
1004
+ "epoch": 77,
1005
+ "loss": 0.2624368397645188,
1006
+ "recon_loss": 0.2150657492561069,
1007
+ "commit_loss": 0.023685545143153932,
1008
+ "embed_loss": 0.023685545143153932,
1009
+ "codebook_utilization": 0.964111328125,
1010
+ "lr": 6.922716010014254e-05,
1011
+ "phase": "full_vq",
1012
+ "val_recon_loss": 0.2176151710949587,
1013
+ "val_utilization": 0.9482421875,
1014
+ "time_sec": 141.2098548412323
1015
+ },
1016
+ {
1017
+ "epoch": 78,
1018
+ "loss": 0.26235106452128426,
1019
+ "recon_loss": 0.21501024715541825,
1020
+ "commit_loss": 0.023670408761174214,
1021
+ "embed_loss": 0.023670408761174214,
1022
+ "codebook_utilization": 0.964111328125,
1023
+ "lr": 6.848110061149553e-05,
1024
+ "phase": "full_vq",
1025
+ "val_recon_loss": 0.21775933932722286,
1026
+ "val_utilization": 0.950927734375,
1027
+ "time_sec": 144.76255917549133
1028
+ },
1029
+ {
1030
+ "epoch": 79,
1031
+ "loss": 0.26233030482117076,
1032
+ "recon_loss": 0.21490754027596012,
1033
+ "commit_loss": 0.023711382371037794,
1034
+ "embed_loss": 0.023711382371037794,
1035
+ "codebook_utilization": 0.964599609375,
1036
+ "lr": 6.773024435212678e-05,
1037
+ "phase": "full_vq",
1038
+ "val_recon_loss": 0.2173820888058523,
1039
+ "val_utilization": 0.95068359375,
1040
+ "time_sec": 141.6652626991272
1041
+ },
1042
+ {
1043
+ "epoch": 80,
1044
+ "loss": 0.2624470307453861,
1045
+ "recon_loss": 0.21483368028793232,
1046
+ "commit_loss": 0.023806675226202986,
1047
+ "embed_loss": 0.023806675226202986,
1048
+ "codebook_utilization": 0.963623046875,
1049
+ "lr": 6.697478620682138e-05,
1050
+ "phase": "full_vq",
1051
+ "val_recon_loss": 0.21697901289784507,
1052
+ "val_utilization": 0.94970703125,
1053
+ "time_sec": 137.51616597175598
1054
+ },
1055
+ {
1056
+ "epoch": 81,
1057
+ "loss": 0.2623364634226331,
1058
+ "recon_loss": 0.21468003019488602,
1059
+ "commit_loss": 0.023828216184808956,
1060
+ "embed_loss": 0.023828216184808956,
1061
+ "codebook_utilization": 0.964111328125,
1062
+ "lr": 6.621492225478415e-05,
1063
+ "phase": "full_vq",
1064
+ "val_recon_loss": 0.21735587850045623,
1065
+ "val_utilization": 0.952880859375,
1066
+ "time_sec": 139.81858944892883
1067
+ },
1068
+ {
1069
+ "epoch": 82,
1070
+ "loss": 0.2622073751679927,
1071
+ "recon_loss": 0.2145599073510829,
1072
+ "commit_loss": 0.023823733850404982,
1073
+ "embed_loss": 0.023823733850404982,
1074
+ "codebook_utilization": 0.964599609375,
1075
+ "lr": 6.545084971874738e-05,
1076
+ "phase": "full_vq",
1077
+ "val_recon_loss": 0.2173672378732917,
1078
+ "val_utilization": 0.94921875,
1079
+ "time_sec": 141.89817309379578
1080
+ },
1081
+ {
1082
+ "epoch": 83,
1083
+ "loss": 0.26239230986533124,
1084
+ "recon_loss": 0.21458652561030736,
1085
+ "commit_loss": 0.023902892011412113,
1086
+ "embed_loss": 0.023902892011412113,
1087
+ "codebook_utilization": 0.963623046875,
1088
+ "lr": 6.468276691378155e-05,
1089
+ "phase": "full_vq",
1090
+ "val_recon_loss": 0.21719316183851006,
1091
+ "val_utilization": 0.95361328125,
1092
+ "time_sec": 140.54402613639832
1093
+ },
1094
+ {
1095
+ "epoch": 84,
1096
+ "loss": 0.2623866770446785,
1097
+ "recon_loss": 0.21454203241885839,
1098
+ "commit_loss": 0.02392232222457323,
1099
+ "embed_loss": 0.02392232222457323,
1100
+ "codebook_utilization": 0.96435546875,
1101
+ "lr": 6.391087319582264e-05,
1102
+ "phase": "full_vq",
1103
+ "val_recon_loss": 0.21700922081644616,
1104
+ "val_utilization": 0.9521484375,
1105
+ "time_sec": 145.78504157066345
1106
+ },
1107
+ {
1108
+ "epoch": 85,
1109
+ "loss": 0.2622778673926343,
1110
+ "recon_loss": 0.21446132791155398,
1111
+ "commit_loss": 0.023908269720348884,
1112
+ "embed_loss": 0.023908269720348884,
1113
+ "codebook_utilization": 0.96435546875,
1114
+ "lr": 6.313536890992936e-05,
1115
+ "phase": "full_vq",
1116
+ "val_recon_loss": 0.21672186431255233,
1117
+ "val_utilization": 0.947021484375,
1118
+ "time_sec": 138.73414516448975
1119
+ },
1120
+ {
1121
+ "epoch": 86,
1122
+ "loss": 0.2620549104441472,
1123
+ "recon_loss": 0.2142903511399822,
1124
+ "commit_loss": 0.023882279255828722,
1125
+ "embed_loss": 0.023882279255828722,
1126
+ "codebook_utilization": 0.96435546875,
1127
+ "lr": 6.23564553382835e-05,
1128
+ "phase": "full_vq",
1129
+ "val_recon_loss": 0.21709384116229047,
1130
+ "val_utilization": 0.9501953125,
1131
+ "time_sec": 136.98708629608154
1132
+ },
1133
+ {
1134
+ "epoch": 87,
1135
+ "loss": 0.2621268547041629,
1136
+ "recon_loss": 0.2142765432555824,
1137
+ "commit_loss": 0.023925155683907712,
1138
+ "embed_loss": 0.023925155683907712,
1139
+ "codebook_utilization": 0.96435546875,
1140
+ "lr": 6.157433464794716e-05,
1141
+ "phase": "full_vq",
1142
+ "val_recon_loss": 0.21668502805608042,
1143
+ "val_utilization": 0.949951171875,
1144
+ "time_sec": 136.89178586006165
1145
+ },
1146
+ {
1147
+ "epoch": 88,
1148
+ "loss": 0.26221330199380555,
1149
+ "recon_loss": 0.21424086038578494,
1150
+ "commit_loss": 0.02398622087467977,
1151
+ "embed_loss": 0.02398622087467977,
1152
+ "codebook_utilization": 0.964111328125,
1153
+ "lr": 6.0789209838390326e-05,
1154
+ "phase": "full_vq",
1155
+ "val_recon_loss": 0.21680285823479128,
1156
+ "val_utilization": 0.952392578125,
1157
+ "time_sec": 146.51975417137146
1158
+ },
1159
+ {
1160
+ "epoch": 89,
1161
+ "loss": 0.2623724174612583,
1162
+ "recon_loss": 0.21422967869980225,
1163
+ "commit_loss": 0.024071369133384892,
1164
+ "embed_loss": 0.024071369133384892,
1165
+ "codebook_utilization": 0.964599609375,
1166
+ "lr": 6.000128468880224e-05,
1167
+ "phase": "full_vq",
1168
+ "val_recon_loss": 0.2167005233382911,
1169
+ "val_utilization": 0.94970703125,
1170
+ "time_sec": 135.16049242019653
1171
+ },
1172
+ {
1173
+ "epoch": 90,
1174
+ "loss": 0.2623601592168575,
1175
+ "recon_loss": 0.2140586848867941,
1176
+ "commit_loss": 0.024150737220557723,
1177
+ "embed_loss": 0.024150737220557723,
1178
+ "codebook_utilization": 0.964111328125,
1179
+ "lr": 5.9210763705200585e-05,
1180
+ "phase": "full_vq",
1181
+ "val_recon_loss": 0.21655598893929062,
1182
+ "val_utilization": 0.95361328125,
1183
+ "time_sec": 139.48371744155884
1184
+ },
1185
+ {
1186
+ "epoch": 91,
1187
+ "loss": 0.26245073332049984,
1188
+ "recon_loss": 0.2140443040749568,
1189
+ "commit_loss": 0.024203214822160358,
1190
+ "embed_loss": 0.024203214822160358,
1191
+ "codebook_utilization": 0.964111328125,
1192
+ "lr": 5.841785206735194e-05,
1193
+ "phase": "full_vq",
1194
+ "val_recon_loss": 0.21645763608511914,
1195
+ "val_utilization": 0.954345703125,
1196
+ "time_sec": 133.07284212112427
1197
+ },
1198
+ {
1199
+ "epoch": 92,
1200
+ "loss": 0.26237547143203455,
1201
+ "recon_loss": 0.21393440621412865,
1202
+ "commit_loss": 0.024220532419659744,
1203
+ "embed_loss": 0.024220532419659744,
1204
+ "codebook_utilization": 0.964111328125,
1205
+ "lr": 5.7622755575517285e-05,
1206
+ "phase": "full_vq",
1207
+ "val_recon_loss": 0.21658130852359064,
1208
+ "val_utilization": 0.95361328125,
1209
+ "time_sec": 141.17808604240417
1210
+ },
1211
+ {
1212
+ "epoch": 93,
1213
+ "loss": 0.26231447358926135,
1214
+ "recon_loss": 0.21383224316648028,
1215
+ "commit_loss": 0.02424111514829281,
1216
+ "embed_loss": 0.02424111514829281,
1217
+ "codebook_utilization": 0.964111328125,
1218
+ "lr": 5.6825680597036586e-05,
1219
+ "phase": "full_vq",
1220
+ "val_recon_loss": 0.2165020809749539,
1221
+ "val_utilization": 0.950927734375,
1222
+ "time_sec": 143.72083640098572
1223
+ },
1224
+ {
1225
+ "epoch": 94,
1226
+ "loss": 0.26243370480653716,
1227
+ "recon_loss": 0.2138815036593737,
1228
+ "commit_loss": 0.024276100679585928,
1229
+ "embed_loss": 0.024276100679585928,
1230
+ "codebook_utilization": 0.964111328125,
1231
+ "lr": 5.602683401276615e-05,
1232
+ "phase": "full_vq",
1233
+ "val_recon_loss": 0.21651848983228877,
1234
+ "val_utilization": 0.95361328125,
1235
+ "time_sec": 143.66276907920837
1236
+ },
1237
+ {
1238
+ "epoch": 95,
1239
+ "loss": 0.26229014348612245,
1240
+ "recon_loss": 0.21371060014174884,
1241
+ "commit_loss": 0.024289771525800065,
1242
+ "embed_loss": 0.024289771525800065,
1243
+ "codebook_utilization": 0.9638671875,
1244
+ "lr": 5.522642316338268e-05,
1245
+ "phase": "full_vq",
1246
+ "val_recon_loss": 0.21645400765236844,
1247
+ "val_utilization": 0.95166015625,
1248
+ "time_sec": 139.4727485179901
1249
+ },
1250
+ {
1251
+ "epoch": 96,
1252
+ "loss": 0.26247825647191947,
1253
+ "recon_loss": 0.21367310435791326,
1254
+ "commit_loss": 0.024402575956046745,
1255
+ "embed_loss": 0.024402575956046745,
1256
+ "codebook_utilization": 0.964111328125,
1257
+ "lr": 5.4424655795567935e-05,
1258
+ "phase": "full_vq",
1259
+ "val_recon_loss": 0.21640119182594705,
1260
+ "val_utilization": 0.95166015625,
1261
+ "time_sec": 143.47048616409302
1262
+ },
1263
+ {
1264
+ "epoch": 97,
1265
+ "loss": 0.2625528493830505,
1266
+ "recon_loss": 0.21364202174713942,
1267
+ "commit_loss": 0.024455413747286086,
1268
+ "embed_loss": 0.024455413747286086,
1269
+ "codebook_utilization": 0.9638671875,
1270
+ "lr": 5.3621740008088126e-05,
1271
+ "phase": "full_vq",
1272
+ "val_recon_loss": 0.21614325490225567,
1273
+ "val_utilization": 0.951416015625,
1274
+ "time_sec": 135.01982259750366
1275
+ },
1276
+ {
1277
+ "epoch": 98,
1278
+ "loss": 0.262200346645624,
1279
+ "recon_loss": 0.21356922713156315,
1280
+ "commit_loss": 0.024315559774697796,
1281
+ "embed_loss": 0.024315559774697796,
1282
+ "codebook_utilization": 0.964111328125,
1283
+ "lr": 5.281788419778186e-05,
1284
+ "phase": "full_vq",
1285
+ "val_recon_loss": 0.21631419332174773,
1286
+ "val_utilization": 0.95166015625,
1287
+ "time_sec": 142.65433263778687
1288
+ },
1289
+ {
1290
+ "epoch": 99,
1291
+ "loss": 0.262547532212605,
1292
+ "recon_loss": 0.21357469924985556,
1293
+ "commit_loss": 0.024486416362750982,
1294
+ "embed_loss": 0.024486416362750982,
1295
+ "codebook_utilization": 0.9638671875,
1296
+ "lr": 5.201329700547076e-05,
1297
+ "phase": "full_vq",
1298
+ "val_recon_loss": 0.216151856639412,
1299
+ "val_utilization": 0.954345703125,
1300
+ "time_sec": 143.2004942893982
1301
+ },
1302
+ {
1303
+ "epoch": 100,
1304
+ "loss": 0.26244740985677173,
1305
+ "recon_loss": 0.21355270792232942,
1306
+ "commit_loss": 0.024447350838501764,
1307
+ "embed_loss": 0.024447350838501764,
1308
+ "codebook_utilization": 0.9638671875,
1309
+ "lr": 5.120818726180663e-05,
1310
+ "phase": "full_vq",
1311
+ "val_recon_loss": 0.21592140105668078,
1312
+ "val_utilization": 0.952392578125,
1313
+ "time_sec": 144.23687529563904
1314
+ },
1315
+ {
1316
+ "epoch": 101,
1317
+ "loss": 0.26259658729965446,
1318
+ "recon_loss": 0.213514934871901,
1319
+ "commit_loss": 0.024540826148255084,
1320
+ "embed_loss": 0.024540826148255084,
1321
+ "codebook_utilization": 0.963623046875,
1322
+ "lr": 5.040276393306951e-05,
1323
+ "phase": "full_vq",
1324
+ "val_recon_loss": 0.21592071786355438,
1325
+ "val_utilization": 0.95263671875,
1326
+ "time_sec": 143.10771584510803
1327
+ },
1328
+ {
1329
+ "epoch": 102,
1330
+ "loss": 0.2624320983684806,
1331
+ "recon_loss": 0.21332257085537845,
1332
+ "commit_loss": 0.02455476401146592,
1333
+ "embed_loss": 0.02455476401146592,
1334
+ "codebook_utilization": 0.9638671875,
1335
+ "lr": 4.9597236066930516e-05,
1336
+ "phase": "full_vq",
1337
+ "val_recon_loss": 0.21663070888666625,
1338
+ "val_utilization": 0.950927734375,
1339
+ "time_sec": 142.21347641944885
1340
+ },
1341
+ {
1342
+ "epoch": 103,
1343
+ "loss": 0.26245626988934306,
1344
+ "recon_loss": 0.21329183280871813,
1345
+ "commit_loss": 0.024582218320732357,
1346
+ "embed_loss": 0.024582218320732357,
1347
+ "codebook_utilization": 0.96337890625,
1348
+ "lr": 4.879181273819339e-05,
1349
+ "phase": "full_vq",
1350
+ "val_recon_loss": 0.21620739276489515,
1351
+ "val_utilization": 0.950927734375,
1352
+ "time_sec": 145.31684947013855
1353
+ },
1354
+ {
1355
+ "epoch": 104,
1356
+ "loss": 0.26249102781700895,
1357
+ "recon_loss": 0.2132447780107418,
1358
+ "commit_loss": 0.024623125009137763,
1359
+ "embed_loss": 0.024623125009137763,
1360
+ "codebook_utilization": 0.9638671875,
1361
+ "lr": 4.798670299452925e-05,
1362
+ "phase": "full_vq",
1363
+ "val_recon_loss": 0.21578157065289744,
1364
+ "val_utilization": 0.94873046875,
1365
+ "time_sec": 143.2276611328125
1366
+ },
1367
+ {
1368
+ "epoch": 105,
1369
+ "loss": 0.26241572643440914,
1370
+ "recon_loss": 0.21322347140974468,
1371
+ "commit_loss": 0.02459612760571687,
1372
+ "embed_loss": 0.02459612760571687,
1373
+ "codebook_utilization": 0.963623046875,
1374
+ "lr": 4.718211580221814e-05,
1375
+ "phase": "full_vq",
1376
+ "val_recon_loss": 0.21590519905760047,
1377
+ "val_utilization": 0.951416015625,
1378
+ "time_sec": 142.68286609649658
1379
+ },
1380
+ {
1381
+ "epoch": 106,
1382
+ "loss": 0.262540693165164,
1383
+ "recon_loss": 0.21309678379678468,
1384
+ "commit_loss": 0.0247219543989879,
1385
+ "embed_loss": 0.0247219543989879,
1386
+ "codebook_utilization": 0.9638671875,
1387
+ "lr": 4.637825999191189e-05,
1388
+ "phase": "full_vq",
1389
+ "val_recon_loss": 0.21581601091984953,
1390
+ "val_utilization": 0.952880859375,
1391
+ "time_sec": 143.06476855278015
1392
+ },
1393
+ {
1394
+ "epoch": 107,
1395
+ "loss": 0.2626226111479245,
1396
+ "recon_loss": 0.21309993232330332,
1397
+ "commit_loss": 0.02476133947288441,
1398
+ "embed_loss": 0.02476133947288441,
1399
+ "codebook_utilization": 0.963623046875,
1400
+ "lr": 4.557534420443209e-05,
1401
+ "phase": "full_vq",
1402
+ "val_recon_loss": 0.21580222692717327,
1403
+ "val_utilization": 0.952880859375,
1404
+ "time_sec": 138.99657583236694
1405
+ },
1406
+ {
1407
+ "epoch": 108,
1408
+ "loss": 0.2625783171030241,
1409
+ "recon_loss": 0.2130398264906917,
1410
+ "commit_loss": 0.02476924516482728,
1411
+ "embed_loss": 0.02476924516482728,
1412
+ "codebook_utilization": 0.9638671875,
1413
+ "lr": 4.477357683661733e-05,
1414
+ "phase": "full_vq",
1415
+ "val_recon_loss": 0.21578819698162294,
1416
+ "val_utilization": 0.94970703125,
1417
+ "time_sec": 141.73647093772888
1418
+ },
1419
+ {
1420
+ "epoch": 109,
1421
+ "loss": 0.2626569550534897,
1422
+ "recon_loss": 0.212990672385628,
1423
+ "commit_loss": 0.024833141578750075,
1424
+ "embed_loss": 0.024833141578750075,
1425
+ "codebook_utilization": 0.9638671875,
1426
+ "lr": 4.397316598723385e-05,
1427
+ "phase": "full_vq",
1428
+ "val_recon_loss": 0.21570578619335476,
1429
+ "val_utilization": 0.95458984375,
1430
+ "time_sec": 143.8497610092163
1431
+ },
1432
+ {
1433
+ "epoch": 110,
1434
+ "loss": 0.26262666285037994,
1435
+ "recon_loss": 0.2129026357801303,
1436
+ "commit_loss": 0.024862013709274572,
1437
+ "embed_loss": 0.024862013709274572,
1438
+ "codebook_utilization": 0.9638671875,
1439
+ "lr": 4.317431940296343e-05,
1440
+ "phase": "full_vq",
1441
+ "val_recon_loss": 0.2154975456468175,
1442
+ "val_utilization": 0.948974609375,
1443
+ "time_sec": 143.44073915481567
1444
+ },
1445
+ {
1446
+ "epoch": 111,
1447
+ "loss": 0.2626328876994167,
1448
+ "recon_loss": 0.21284882169025054,
1449
+ "commit_loss": 0.024892033029822153,
1450
+ "embed_loss": 0.024892033029822153,
1451
+ "codebook_utilization": 0.9638671875,
1452
+ "lr": 4.237724442448274e-05,
1453
+ "phase": "full_vq",
1454
+ "val_recon_loss": 0.21570696514309123,
1455
+ "val_utilization": 0.94873046875,
1456
+ "time_sec": 138.77132177352905
1457
+ },
1458
+ {
1459
+ "epoch": 112,
1460
+ "loss": 0.2625980107441827,
1461
+ "recon_loss": 0.21283445339700394,
1462
+ "commit_loss": 0.024881778650874205,
1463
+ "embed_loss": 0.024881778650874205,
1464
+ "codebook_utilization": 0.96337890625,
1465
+ "lr": 4.158214793264808e-05,
1466
+ "phase": "full_vq",
1467
+ "val_recon_loss": 0.21572973378253787,
1468
+ "val_utilization": 0.95361328125,
1469
+ "time_sec": 143.36263871192932
1470
+ },
1471
+ {
1472
+ "epoch": 113,
1473
+ "loss": 0.2627456883185601,
1474
+ "recon_loss": 0.2128262622610017,
1475
+ "commit_loss": 0.02495971315245076,
1476
+ "embed_loss": 0.02495971315245076,
1477
+ "codebook_utilization": 0.9638671875,
1478
+ "lr": 4.078923629479945e-05,
1479
+ "phase": "full_vq",
1480
+ "val_recon_loss": 0.2162084909302465,
1481
+ "val_utilization": 0.947998046875,
1482
+ "time_sec": 139.92266058921814
1483
+ },
1484
+ {
1485
+ "epoch": 114,
1486
+ "loss": 0.262728397600696,
1487
+ "recon_loss": 0.21278782003132632,
1488
+ "commit_loss": 0.024970288842734768,
1489
+ "embed_loss": 0.024970288842734768,
1490
+ "codebook_utilization": 0.9638671875,
1491
+ "lr": 3.9998715311197785e-05,
1492
+ "phase": "full_vq",
1493
+ "val_recon_loss": 0.2156060559026311,
1494
+ "val_utilization": 0.950439453125,
1495
+ "time_sec": 140.2749536037445
1496
+ },
1497
+ {
1498
+ "epoch": 115,
1499
+ "loss": 0.26264743426143317,
1500
+ "recon_loss": 0.21265302865001245,
1501
+ "commit_loss": 0.02499720252050859,
1502
+ "embed_loss": 0.02499720252050859,
1503
+ "codebook_utilization": 0.9638671875,
1504
+ "lr": 3.9210790161609705e-05,
1505
+ "phase": "full_vq",
1506
+ "val_recon_loss": 0.2153637885545077,
1507
+ "val_utilization": 0.95068359375,
1508
+ "time_sec": 139.71399354934692
1509
+ },
1510
+ {
1511
+ "epoch": 116,
1512
+ "loss": 0.2626770178719265,
1513
+ "recon_loss": 0.21262222270009318,
1514
+ "commit_loss": 0.02502739784335541,
1515
+ "embed_loss": 0.02502739784335541,
1516
+ "codebook_utilization": 0.96337890625,
1517
+ "lr": 3.8425665352052864e-05,
1518
+ "phase": "full_vq",
1519
+ "val_recon_loss": 0.2154450115193142,
1520
+ "val_utilization": 0.951904296875,
1521
+ "time_sec": 134.02279019355774
1522
+ },
1523
+ {
1524
+ "epoch": 117,
1525
+ "loss": 0.26282290857901097,
1526
+ "recon_loss": 0.21260020419511044,
1527
+ "commit_loss": 0.02511135214651988,
1528
+ "embed_loss": 0.02511135214651988,
1529
+ "codebook_utilization": 0.9638671875,
1530
+ "lr": 3.764354466171653e-05,
1531
+ "phase": "full_vq",
1532
+ "val_recon_loss": 0.21545669796426645,
1533
+ "val_utilization": 0.95263671875,
1534
+ "time_sec": 136.37731957435608
1535
+ },
1536
+ {
1537
+ "epoch": 118,
1538
+ "loss": 0.26262964233554154,
1539
+ "recon_loss": 0.2125766294071991,
1540
+ "commit_loss": 0.02502650658784278,
1541
+ "embed_loss": 0.02502650658784278,
1542
+ "codebook_utilization": 0.96337890625,
1543
+ "lr": 3.686463109007067e-05,
1544
+ "phase": "full_vq",
1545
+ "val_recon_loss": 0.21550561554646225,
1546
+ "val_utilization": 0.95263671875,
1547
+ "time_sec": 138.2373538017273
1548
+ },
1549
+ {
1550
+ "epoch": 119,
1551
+ "loss": 0.26272081553451415,
1552
+ "recon_loss": 0.21248449017684956,
1553
+ "commit_loss": 0.025118162666212737,
1554
+ "embed_loss": 0.025118162666212737,
1555
+ "codebook_utilization": 0.963623046875,
1556
+ "lr": 3.6089126804177374e-05,
1557
+ "phase": "full_vq",
1558
+ "val_recon_loss": 0.21534196170193426,
1559
+ "val_utilization": 0.954345703125,
1560
+ "time_sec": 133.23958468437195
1561
+ },
1562
+ {
1563
+ "epoch": 120,
1564
+ "loss": 0.2629021041317361,
1565
+ "recon_loss": 0.21248677829255255,
1566
+ "commit_loss": 0.025207663012976406,
1567
+ "embed_loss": 0.025207663012976406,
1568
+ "codebook_utilization": 0.963623046875,
1569
+ "lr": 3.531723308621847e-05,
1570
+ "phase": "full_vq",
1571
+ "val_recon_loss": 0.21530195526527554,
1572
+ "val_utilization": 0.95263671875,
1573
+ "time_sec": 134.80960869789124
1574
+ },
1575
+ {
1576
+ "epoch": 121,
1577
+ "loss": 0.26279964883556856,
1578
+ "recon_loss": 0.21241148916605687,
1579
+ "commit_loss": 0.025194079779229835,
1580
+ "embed_loss": 0.025194079779229835,
1581
+ "codebook_utilization": 0.963623046875,
1582
+ "lr": 3.4549150281252636e-05,
1583
+ "phase": "full_vq",
1584
+ "val_recon_loss": 0.21515031317981442,
1585
+ "val_utilization": 0.950439453125,
1586
+ "time_sec": 134.04538369178772
1587
+ },
1588
+ {
1589
+ "epoch": 122,
1590
+ "loss": 0.2628727775359864,
1591
+ "recon_loss": 0.21235055496618355,
1592
+ "commit_loss": 0.025261111297521004,
1593
+ "embed_loss": 0.025261111297521004,
1594
+ "codebook_utilization": 0.963623046875,
1595
+ "lr": 3.3785077745215873e-05,
1596
+ "phase": "full_vq",
1597
+ "val_recon_loss": 0.21535600906007746,
1598
+ "val_utilization": 0.95166015625,
1599
+ "time_sec": 135.19442105293274
1600
+ },
1601
+ {
1602
+ "epoch": 123,
1603
+ "loss": 0.26274573273936586,
1604
+ "recon_loss": 0.21237245981447742,
1605
+ "commit_loss": 0.025186636611354868,
1606
+ "embed_loss": 0.025186636611354868,
1607
+ "codebook_utilization": 0.963134765625,
1608
+ "lr": 3.302521379317865e-05,
1609
+ "phase": "full_vq",
1610
+ "val_recon_loss": 0.21540050652254833,
1611
+ "val_utilization": 0.95068359375,
1612
+ "time_sec": 145.54879689216614
1613
+ },
1614
+ {
1615
+ "epoch": 124,
1616
+ "loss": 0.26281195085465425,
1617
+ "recon_loss": 0.21222630131050824,
1618
+ "commit_loss": 0.02529282482759901,
1619
+ "embed_loss": 0.02529282482759901,
1620
+ "codebook_utilization": 0.963623046875,
1621
+ "lr": 3.226975564787323e-05,
1622
+ "phase": "full_vq",
1623
+ "val_recon_loss": 0.2151435732506634,
1624
+ "val_utilization": 0.952880859375,
1625
+ "time_sec": 140.79991722106934
1626
+ },
1627
+ {
1628
+ "epoch": 125,
1629
+ "loss": 0.26292738584968134,
1630
+ "recon_loss": 0.21221060398713684,
1631
+ "commit_loss": 0.025358390641022667,
1632
+ "embed_loss": 0.025358390641022667,
1633
+ "codebook_utilization": 0.963623046875,
1634
+ "lr": 3.151889938850445e-05,
1635
+ "phase": "full_vq",
1636
+ "val_recon_loss": 0.21514364013846002,
1637
+ "val_utilization": 0.952880859375,
1638
+ "time_sec": 136.39927530288696
1639
+ },
1640
+ {
1641
+ "epoch": 126,
1642
+ "loss": 0.2628670044263527,
1643
+ "recon_loss": 0.21219596272803903,
1644
+ "commit_loss": 0.0253355209475893,
1645
+ "embed_loss": 0.0253355209475893,
1646
+ "codebook_utilization": 0.963623046875,
1647
+ "lr": 3.0772839899857464e-05,
1648
+ "phase": "full_vq",
1649
+ "val_recon_loss": 0.21489372648549884,
1650
+ "val_utilization": 0.95166015625,
1651
+ "time_sec": 136.2484495639801
1652
+ },
1653
+ {
1654
+ "epoch": 127,
1655
+ "loss": 0.2628906367593988,
1656
+ "recon_loss": 0.21222007724893124,
1657
+ "commit_loss": 0.025335279699707742,
1658
+ "embed_loss": 0.025335279699707742,
1659
+ "codebook_utilization": 0.963134765625,
1660
+ "lr": 3.003177082171522e-05,
1661
+ "phase": "full_vq",
1662
+ "val_recon_loss": 0.21487673966402418,
1663
+ "val_utilization": 0.950439453125,
1664
+ "time_sec": 142.29378175735474
1665
+ },
1666
+ {
1667
+ "epoch": 128,
1668
+ "loss": 0.2629637006862054,
1669
+ "recon_loss": 0.21212804834371013,
1670
+ "commit_loss": 0.025417826188914982,
1671
+ "embed_loss": 0.025417826188914982,
1672
+ "codebook_utilization": 0.963623046875,
1673
+ "lr": 2.92958844985994e-05,
1674
+ "phase": "full_vq",
1675
+ "val_recon_loss": 0.21535063509860736,
1676
+ "val_utilization": 0.947509765625,
1677
+ "time_sec": 140.40249490737915
1678
+ },
1679
+ {
1680
+ "epoch": 129,
1681
+ "loss": 0.2629646127664946,
1682
+ "recon_loss": 0.2121002840478892,
1683
+ "commit_loss": 0.025432164038766206,
1684
+ "embed_loss": 0.025432164038766206,
1685
+ "codebook_utilization": 0.96337890625,
1686
+ "lr": 2.8565371929847304e-05,
1687
+ "phase": "full_vq",
1688
+ "val_recon_loss": 0.21501677274034264,
1689
+ "val_utilization": 0.95068359375,
1690
+ "time_sec": 144.31898498535156
1691
+ },
1692
+ {
1693
+ "epoch": 130,
1694
+ "loss": 0.26299205032709816,
1695
+ "recon_loss": 0.21205659505474536,
1696
+ "commit_loss": 0.025467727694226314,
1697
+ "embed_loss": 0.025467727694226314,
1698
+ "codebook_utilization": 0.96337890625,
1699
+ "lr": 2.7840422720037937e-05,
1700
+ "phase": "full_vq",
1701
+ "val_recon_loss": 0.21497032841604746,
1702
+ "val_utilization": 0.95263671875,
1703
+ "time_sec": 139.77696990966797
1704
+ },
1705
+ {
1706
+ "epoch": 131,
1707
+ "loss": 0.2631795630868534,
1708
+ "recon_loss": 0.21205847801231756,
1709
+ "commit_loss": 0.025560542562507033,
1710
+ "embed_loss": 0.025560542562507033,
1711
+ "codebook_utilization": 0.963134765625,
1712
+ "lr": 2.7121225029780234e-05,
1713
+ "phase": "full_vq",
1714
+ "val_recon_loss": 0.21494582521446634,
1715
+ "val_utilization": 0.95166015625,
1716
+ "time_sec": 136.54157423973083
1717
+ },
1718
+ {
1719
+ "epoch": 132,
1720
+ "loss": 0.26304175452326695,
1721
+ "recon_loss": 0.21195026341734863,
1722
+ "commit_loss": 0.025545745780111005,
1723
+ "embed_loss": 0.025545745780111005,
1724
+ "codebook_utilization": 0.963134765625,
1725
+ "lr": 2.6407965526875892e-05,
1726
+ "phase": "full_vq",
1727
+ "val_recon_loss": 0.21491514347242505,
1728
+ "val_utilization": 0.95166015625,
1729
+ "time_sec": 144.39372777938843
1730
+ },
1731
+ {
1732
+ "epoch": 133,
1733
+ "loss": 0.2631893021465963,
1734
+ "recon_loss": 0.21197636837397166,
1735
+ "commit_loss": 0.025606466800499415,
1736
+ "embed_loss": 0.025606466800499415,
1737
+ "codebook_utilization": 0.963134765625,
1738
+ "lr": 2.570082933786969e-05,
1739
+ "phase": "full_vq",
1740
+ "val_recon_loss": 0.21499076912577234,
1741
+ "val_utilization": 0.950927734375,
1742
+ "time_sec": 144.53392052650452
1743
+ },
1744
+ {
1745
+ "epoch": 134,
1746
+ "loss": 0.262954132606182,
1747
+ "recon_loss": 0.21191143183931102,
1748
+ "commit_loss": 0.0255213502799552,
1749
+ "embed_loss": 0.0255213502799552,
1750
+ "codebook_utilization": 0.96337890625,
1751
+ "lr": 2.5000000000000008e-05,
1752
+ "phase": "full_vq",
1753
+ "val_recon_loss": 0.21492530388778516,
1754
+ "val_utilization": 0.950927734375,
1755
+ "time_sec": 138.9941623210907
1756
+ },
1757
+ {
1758
+ "epoch": 135,
1759
+ "loss": 0.2631179538926458,
1760
+ "recon_loss": 0.21188214994704496,
1761
+ "commit_loss": 0.025617901866796217,
1762
+ "embed_loss": 0.025617901866796217,
1763
+ "codebook_utilization": 0.96337890625,
1764
+ "lr": 2.4305659413561568e-05,
1765
+ "phase": "full_vq",
1766
+ "val_recon_loss": 0.21498684752523228,
1767
+ "val_utilization": 0.952880859375,
1768
+ "time_sec": 137.7155773639679
1769
+ },
1770
+ {
1771
+ "epoch": 136,
1772
+ "loss": 0.263213415456012,
1773
+ "recon_loss": 0.2118592612182868,
1774
+ "commit_loss": 0.025677077247581993,
1775
+ "embed_loss": 0.025677077247581993,
1776
+ "codebook_utilization": 0.96337890625,
1777
+ "lr": 2.3617987794693356e-05,
1778
+ "phase": "full_vq",
1779
+ "val_recon_loss": 0.21508604455529973,
1780
+ "val_utilization": 0.951416015625,
1781
+ "time_sec": 136.0383710861206
1782
+ },
1783
+ {
1784
+ "epoch": 137,
1785
+ "loss": 0.26307226824485835,
1786
+ "recon_loss": 0.21179428300883388,
1787
+ "commit_loss": 0.025638992703825154,
1788
+ "embed_loss": 0.025638992703825154,
1789
+ "codebook_utilization": 0.96337890625,
1790
+ "lr": 2.2937163628603435e-05,
1791
+ "phase": "full_vq",
1792
+ "val_recon_loss": 0.21478415941924192,
1793
+ "val_utilization": 0.9501953125,
1794
+ "time_sec": 140.13938450813293
1795
+ },
1796
+ {
1797
+ "epoch": 138,
1798
+ "loss": 0.2632945872298101,
1799
+ "recon_loss": 0.21177088443018233,
1800
+ "commit_loss": 0.02576185122061632,
1801
+ "embed_loss": 0.02576185122061632,
1802
+ "codebook_utilization": 0.963134765625,
1803
+ "lr": 2.2263363623243054e-05,
1804
+ "phase": "full_vq",
1805
+ "val_recon_loss": 0.21485859306340807,
1806
+ "val_utilization": 0.951416015625,
1807
+ "time_sec": 134.60354804992676
1808
+ },
1809
+ {
1810
+ "epoch": 139,
1811
+ "loss": 0.26323897452131517,
1812
+ "recon_loss": 0.21172283070924158,
1813
+ "commit_loss": 0.025758072006993177,
1814
+ "embed_loss": 0.025758072006993177,
1815
+ "codebook_utilization": 0.963134765625,
1816
+ "lr": 2.1596762663442214e-05,
1817
+ "phase": "full_vq",
1818
+ "val_recon_loss": 0.21481114718016614,
1819
+ "val_utilization": 0.94921875,
1820
+ "time_sec": 142.61455631256104
1821
+ },
1822
+ {
1823
+ "epoch": 140,
1824
+ "loss": 0.26323034293283293,
1825
+ "recon_loss": 0.2116950522189541,
1826
+ "commit_loss": 0.02576764571028673,
1827
+ "embed_loss": 0.02576764571028673,
1828
+ "codebook_utilization": 0.963134765625,
1829
+ "lr": 2.0937533765518167e-05,
1830
+ "phase": "full_vq",
1831
+ "val_recon_loss": 0.21475737257285066,
1832
+ "val_utilization": 0.951904296875,
1833
+ "time_sec": 136.42186188697815
1834
+ },
1835
+ {
1836
+ "epoch": 141,
1837
+ "loss": 0.26327095095946534,
1838
+ "recon_loss": 0.2116691388371514,
1839
+ "commit_loss": 0.0258009060132027,
1840
+ "embed_loss": 0.0258009060132027,
1841
+ "codebook_utilization": 0.963134765625,
1842
+ "lr": 2.0285848032369133e-05,
1843
+ "phase": "full_vq",
1844
+ "val_recon_loss": 0.21490554009260757,
1845
+ "val_utilization": 0.949951171875,
1846
+ "time_sec": 134.38898229599
1847
+ },
1848
+ {
1849
+ "epoch": 142,
1850
+ "loss": 0.2632362805407868,
1851
+ "recon_loss": 0.2116224230304966,
1852
+ "commit_loss": 0.025806928916675288,
1853
+ "embed_loss": 0.025806928916675288,
1854
+ "codebook_utilization": 0.962890625,
1855
+ "lr": 1.964187460906444e-05,
1856
+ "phase": "full_vq",
1857
+ "val_recon_loss": 0.2146551665965091,
1858
+ "val_utilization": 0.953125,
1859
+ "time_sec": 133.89641070365906
1860
+ },
1861
+ {
1862
+ "epoch": 143,
1863
+ "loss": 0.263446235620394,
1864
+ "recon_loss": 0.2116917203152729,
1865
+ "commit_loss": 0.025877257955429677,
1866
+ "embed_loss": 0.025877257955429677,
1867
+ "codebook_utilization": 0.962890625,
1868
+ "lr": 1.9005780638942995e-05,
1869
+ "phase": "full_vq",
1870
+ "val_recon_loss": 0.21470123839177443,
1871
+ "val_utilization": 0.952392578125,
1872
+ "time_sec": 134.22949767112732
1873
+ },
1874
+ {
1875
+ "epoch": 144,
1876
+ "loss": 0.26336621955481326,
1877
+ "recon_loss": 0.21160294533502766,
1878
+ "commit_loss": 0.025881637347140288,
1879
+ "embed_loss": 0.025881637347140288,
1880
+ "codebook_utilization": 0.962890625,
1881
+ "lr": 1.837773122023114e-05,
1882
+ "phase": "full_vq",
1883
+ "val_recon_loss": 0.214663444777553,
1884
+ "val_utilization": 0.950439453125,
1885
+ "time_sec": 134.90559148788452
1886
+ },
1887
+ {
1888
+ "epoch": 145,
1889
+ "loss": 0.26333468379124714,
1890
+ "recon_loss": 0.21154510017817582,
1891
+ "commit_loss": 0.025894791834298672,
1892
+ "embed_loss": 0.025894791834298672,
1893
+ "codebook_utilization": 0.963134765625,
1894
+ "lr": 1.775788936319148e-05,
1895
+ "phase": "full_vq",
1896
+ "val_recon_loss": 0.21469675916969105,
1897
+ "val_utilization": 0.950439453125,
1898
+ "time_sec": 135.05161809921265
1899
+ },
1900
+ {
1901
+ "epoch": 146,
1902
+ "loss": 0.26336115425033624,
1903
+ "recon_loss": 0.21155201682068792,
1904
+ "commit_loss": 0.02590456905302801,
1905
+ "embed_loss": 0.02590456905302801,
1906
+ "codebook_utilization": 0.963134765625,
1907
+ "lr": 1.714641594781347e-05,
1908
+ "phase": "full_vq",
1909
+ "val_recon_loss": 0.21472269999846982,
1910
+ "val_utilization": 0.951904296875,
1911
+ "time_sec": 133.29162096977234
1912
+ },
1913
+ {
1914
+ "epoch": 147,
1915
+ "loss": 0.2635334265748983,
1916
+ "recon_loss": 0.2115408958902527,
1917
+ "commit_loss": 0.025996265284272874,
1918
+ "embed_loss": 0.025996265284272874,
1919
+ "codebook_utilization": 0.963134765625,
1920
+ "lr": 1.654346968205709e-05,
1921
+ "phase": "full_vq",
1922
+ "val_recon_loss": 0.21469566920834982,
1923
+ "val_utilization": 0.95166015625,
1924
+ "time_sec": 136.36484789848328
1925
+ },
1926
+ {
1927
+ "epoch": 148,
1928
+ "loss": 0.26344648839496987,
1929
+ "recon_loss": 0.21149396928668346,
1930
+ "commit_loss": 0.025976259700529945,
1931
+ "embed_loss": 0.025976259700529945,
1932
+ "codebook_utilization": 0.962890625,
1933
+ "lr": 1.5949207060660148e-05,
1934
+ "phase": "full_vq",
1935
+ "val_recon_loss": 0.2145408165923665,
1936
+ "val_utilization": 0.951416015625,
1937
+ "time_sec": 135.17042112350464
1938
+ },
1939
+ {
1940
+ "epoch": 149,
1941
+ "loss": 0.2635232643665983,
1942
+ "recon_loss": 0.21148281600982516,
1943
+ "commit_loss": 0.026020224347488505,
1944
+ "embed_loss": 0.026020224347488505,
1945
+ "codebook_utilization": 0.962646484375,
1946
+ "lr": 1.536378232452003e-05,
1947
+ "phase": "full_vq",
1948
+ "val_recon_loss": 0.2145494936724727,
1949
+ "val_utilization": 0.950927734375,
1950
+ "time_sec": 139.20989346504211
1951
+ },
1952
+ {
1953
+ "epoch": 150,
1954
+ "loss": 0.26356390633476456,
1955
+ "recon_loss": 0.21145685033261938,
1956
+ "commit_loss": 0.026053527953118327,
1957
+ "embed_loss": 0.026053527953118327,
1958
+ "codebook_utilization": 0.962646484375,
1959
+ "lr": 1.478734742066054e-05,
1960
+ "phase": "full_vq",
1961
+ "val_recon_loss": 0.2145248321669825,
1962
+ "val_utilization": 0.950927734375,
1963
+ "time_sec": 134.27815127372742
1964
+ },
1965
+ {
1966
+ "epoch": 151,
1967
+ "loss": 0.263456537976336,
1968
+ "recon_loss": 0.21137746762696322,
1969
+ "commit_loss": 0.02603953535388398,
1970
+ "embed_loss": 0.02603953535388398,
1971
+ "codebook_utilization": 0.96240234375,
1972
+ "lr": 1.4220051962793949e-05,
1973
+ "phase": "full_vq",
1974
+ "val_recon_loss": 0.21460410157281362,
1975
+ "val_utilization": 0.951171875,
1976
+ "time_sec": 147.3650257587433
1977
+ },
1978
+ {
1979
+ "epoch": 152,
1980
+ "loss": 0.2635752874297824,
1981
+ "recon_loss": 0.2114138630468671,
1982
+ "commit_loss": 0.026080712345416146,
1983
+ "embed_loss": 0.026080712345416146,
1984
+ "codebook_utilization": 0.962646484375,
1985
+ "lr": 1.3662043192488835e-05,
1986
+ "phase": "full_vq",
1987
+ "val_recon_loss": 0.21458580562572802,
1988
+ "val_utilization": 0.951171875,
1989
+ "time_sec": 137.05629324913025
1990
+ },
1991
+ {
1992
+ "epoch": 153,
1993
+ "loss": 0.26350862191621527,
1994
+ "recon_loss": 0.2113756863324623,
1995
+ "commit_loss": 0.02606646752686601,
1996
+ "embed_loss": 0.02606646752686601,
1997
+ "codebook_utilization": 0.962646484375,
1998
+ "lr": 1.3113465940953492e-05,
1999
+ "phase": "full_vq",
2000
+ "val_recon_loss": 0.2145426593972056,
2001
+ "val_utilization": 0.95166015625,
2002
+ "time_sec": 132.64040899276733
2003
+ },
2004
+ {
2005
+ "epoch": 154,
2006
+ "loss": 0.263515235144433,
2007
+ "recon_loss": 0.21134059215948833,
2008
+ "commit_loss": 0.026087321661574247,
2009
+ "embed_loss": 0.026087321661574247,
2010
+ "codebook_utilization": 0.962646484375,
2011
+ "lr": 1.2574462591444937e-05,
2012
+ "phase": "full_vq",
2013
+ "val_recon_loss": 0.2145483261078931,
2014
+ "val_utilization": 0.949462890625,
2015
+ "time_sec": 140.30550003051758
2016
+ },
2017
+ {
2018
+ "epoch": 155,
2019
+ "loss": 0.263574954677565,
2020
+ "recon_loss": 0.21133515969686067,
2021
+ "commit_loss": 0.02611989762411935,
2022
+ "embed_loss": 0.02611989762411935,
2023
+ "codebook_utilization": 0.96240234375,
2024
+ "lr": 1.2045173042313425e-05,
2025
+ "phase": "full_vq",
2026
+ "val_recon_loss": 0.21447547541910342,
2027
+ "val_utilization": 0.951416015625,
2028
+ "time_sec": 138.4757435321808
2029
+ },
2030
+ {
2031
+ "epoch": 156,
2032
+ "loss": 0.263707318542611,
2033
+ "recon_loss": 0.21133486664069054,
2034
+ "commit_loss": 0.02618622616801643,
2035
+ "embed_loss": 0.02618622616801643,
2036
+ "codebook_utilization": 0.96240234375,
2037
+ "lr": 1.1525734670691714e-05,
2038
+ "phase": "full_vq",
2039
+ "val_recon_loss": 0.2145009935739335,
2040
+ "val_utilization": 0.950439453125,
2041
+ "time_sec": 141.199223279953
2042
+ },
2043
+ {
2044
+ "epoch": 157,
2045
+ "loss": 0.2635939145879694,
2046
+ "recon_loss": 0.21131640018486395,
2047
+ "commit_loss": 0.02613875730755691,
2048
+ "embed_loss": 0.02613875730755691,
2049
+ "codebook_utilization": 0.962646484375,
2050
+ "lr": 1.1016282296838872e-05,
2051
+ "phase": "full_vq",
2052
+ "val_recon_loss": 0.21450917548343038,
2053
+ "val_utilization": 0.951904296875,
2054
+ "time_sec": 136.1280906200409
2055
+ },
2056
+ {
2057
+ "epoch": 158,
2058
+ "loss": 0.26378513707054985,
2059
+ "recon_loss": 0.21128644679135422,
2060
+ "commit_loss": 0.026249345079023986,
2061
+ "embed_loss": 0.026249345079023986,
2062
+ "codebook_utilization": 0.96240234375,
2063
+ "lr": 1.051694814914775e-05,
2064
+ "phase": "full_vq",
2065
+ "val_recon_loss": 0.21447015896941837,
2066
+ "val_utilization": 0.951416015625,
2067
+ "time_sec": 140.01632690429688
2068
+ },
2069
+ {
2070
+ "epoch": 159,
2071
+ "loss": 0.2637370531838438,
2072
+ "recon_loss": 0.2112936730872648,
2073
+ "commit_loss": 0.026221690048289493,
2074
+ "embed_loss": 0.026221690048289493,
2075
+ "codebook_utilization": 0.962158203125,
2076
+ "lr": 1.002786182982495e-05,
2077
+ "phase": "full_vq",
2078
+ "val_recon_loss": 0.21447608201356416,
2079
+ "val_utilization": 0.95166015625,
2080
+ "time_sec": 136.5990710258484
2081
+ },
2082
+ {
2083
+ "epoch": 160,
2084
+ "loss": 0.26371950195814536,
2085
+ "recon_loss": 0.2112749334232917,
2086
+ "commit_loss": 0.026222284348191932,
2087
+ "embed_loss": 0.026222284348191932,
2088
+ "codebook_utilization": 0.962158203125,
2089
+ "lr": 9.549150281252631e-06,
2090
+ "phase": "full_vq",
2091
+ "val_recon_loss": 0.21466263316655426,
2092
+ "val_utilization": 0.950439453125,
2093
+ "time_sec": 137.78337717056274
2094
+ },
2095
+ {
2096
+ "epoch": 161,
2097
+ "loss": 0.26380349861250985,
2098
+ "recon_loss": 0.21125120526245292,
2099
+ "commit_loss": 0.026276146536213435,
2100
+ "embed_loss": 0.026276146536213435,
2101
+ "codebook_utilization": 0.962158203125,
2102
+ "lr": 9.080937753040662e-06,
2103
+ "phase": "full_vq",
2104
+ "val_recon_loss": 0.2144539835077993,
2105
+ "val_utilization": 0.95166015625,
2106
+ "time_sec": 136.94650840759277
2107
+ },
2108
+ {
2109
+ "epoch": 162,
2110
+ "loss": 0.2637765736680044,
2111
+ "recon_loss": 0.2112805433066557,
2112
+ "commit_loss": 0.026248015152911346,
2113
+ "embed_loss": 0.026248015152911346,
2114
+ "codebook_utilization": 0.962158203125,
2115
+ "lr": 8.62334576977751e-06,
2116
+ "phase": "full_vq",
2117
+ "val_recon_loss": 0.2144657684008727,
2118
+ "val_utilization": 0.951904296875,
2119
+ "time_sec": 138.99861001968384
2120
+ },
2121
+ {
2122
+ "epoch": 163,
2123
+ "loss": 0.26381653016256446,
2124
+ "recon_loss": 0.21117154924769388,
2125
+ "commit_loss": 0.026322490636632855,
2126
+ "embed_loss": 0.026322490636632855,
2127
+ "codebook_utilization": 0.962158203125,
2128
+ "lr": 8.176493099488661e-06,
2129
+ "phase": "full_vq",
2130
+ "val_recon_loss": 0.21452430418033278,
2131
+ "val_utilization": 0.9521484375,
2132
+ "time_sec": 138.35024857521057
2133
+ },
2134
+ {
2135
+ "epoch": 164,
2136
+ "loss": 0.26380788290565255,
2137
+ "recon_loss": 0.21118690212809943,
2138
+ "commit_loss": 0.026310490247437625,
2139
+ "embed_loss": 0.026310490247437625,
2140
+ "codebook_utilization": 0.962158203125,
2141
+ "lr": 7.740495722810268e-06,
2142
+ "phase": "full_vq",
2143
+ "val_recon_loss": 0.21444993189881356,
2144
+ "val_utilization": 0.951416015625,
2145
+ "time_sec": 140.16536498069763
2146
+ },
2147
+ {
2148
+ "epoch": 165,
2149
+ "loss": 0.26384868961443425,
2150
+ "recon_loss": 0.21120170037436292,
2151
+ "commit_loss": 0.026323494458505455,
2152
+ "embed_loss": 0.026323494458505455,
2153
+ "codebook_utilization": 0.962158203125,
2154
+ "lr": 7.315466802886386e-06,
2155
+ "phase": "full_vq",
2156
+ "val_recon_loss": 0.21445591145017173,
2157
+ "val_utilization": 0.9521484375,
2158
+ "time_sec": 140.13656520843506
2159
+ },
2160
+ {
2161
+ "epoch": 166,
2162
+ "loss": 0.2639256482767219,
2163
+ "recon_loss": 0.21115755903607786,
2164
+ "commit_loss": 0.026384044971145427,
2165
+ "embed_loss": 0.026384044971145427,
2166
+ "codebook_utilization": 0.962158203125,
2167
+ "lr": 6.901516655997545e-06,
2168
+ "phase": "full_vq",
2169
+ "val_recon_loss": 0.21443970066108062,
2170
+ "val_utilization": 0.952392578125,
2171
+ "time_sec": 144.7363474369049
2172
+ },
2173
+ {
2174
+ "epoch": 167,
2175
+ "loss": 0.2638934272818449,
2176
+ "recon_loss": 0.21113231188559597,
2177
+ "commit_loss": 0.02638055769307665,
2178
+ "embed_loss": 0.02638055769307665,
2179
+ "codebook_utilization": 0.962158203125,
2180
+ "lr": 6.4987527229280395e-06,
2181
+ "phase": "full_vq",
2182
+ "val_recon_loss": 0.2144314633326584,
2183
+ "val_utilization": 0.95166015625,
2184
+ "time_sec": 147.2448627948761
2185
+ },
2186
+ {
2187
+ "epoch": 168,
2188
+ "loss": 0.26390171210536467,
2189
+ "recon_loss": 0.21111351643312914,
2190
+ "commit_loss": 0.026394098058221786,
2191
+ "embed_loss": 0.026394098058221786,
2192
+ "codebook_utilization": 0.962158203125,
2193
+ "lr": 6.107279541079777e-06,
2194
+ "phase": "full_vq",
2195
+ "val_recon_loss": 0.21441773107547438,
2196
+ "val_utilization": 0.951904296875,
2197
+ "time_sec": 144.23006176948547
2198
+ },
2199
+ {
2200
+ "epoch": 169,
2201
+ "loss": 0.2639025990475161,
2202
+ "recon_loss": 0.21107478567012927,
2203
+ "commit_loss": 0.026413906819936706,
2204
+ "embed_loss": 0.026413906819936706,
2205
+ "codebook_utilization": 0.962158203125,
2206
+ "lr": 5.727198717339518e-06,
2207
+ "phase": "full_vq",
2208
+ "val_recon_loss": 0.21439227327871857,
2209
+ "val_utilization": 0.95068359375,
2210
+ "time_sec": 143.63215351104736
2211
+ },
2212
+ {
2213
+ "epoch": 170,
2214
+ "loss": 0.26404363062323594,
2215
+ "recon_loss": 0.21114839552863826,
2216
+ "commit_loss": 0.026447617544774926,
2217
+ "embed_loss": 0.026447617544774926,
2218
+ "codebook_utilization": 0.962158203125,
2219
+ "lr": 5.358608901706794e-06,
2220
+ "phase": "full_vq",
2221
+ "val_recon_loss": 0.2144841035262922,
2222
+ "val_utilization": 0.951904296875,
2223
+ "time_sec": 142.67631769180298
2224
+ },
2225
+ {
2226
+ "epoch": 171,
2227
+ "loss": 0.2639528940807836,
2228
+ "recon_loss": 0.2111168358384109,
2229
+ "commit_loss": 0.026418029456866304,
2230
+ "embed_loss": 0.026418029456866304,
2231
+ "codebook_utilization": 0.962158203125,
2232
+ "lr": 5.001605761689385e-06,
2233
+ "phase": "full_vq",
2234
+ "val_recon_loss": 0.21443519029724464,
2235
+ "val_utilization": 0.951904296875,
2236
+ "time_sec": 134.74238967895508
2237
+ },
2238
+ {
2239
+ "epoch": 172,
2240
+ "loss": 0.26402379063571374,
2241
+ "recon_loss": 0.21109858038141152,
2242
+ "commit_loss": 0.02646260511705546,
2243
+ "embed_loss": 0.02646260511705546,
2244
+ "codebook_utilization": 0.962158203125,
2245
+ "lr": 4.656281957472728e-06,
2246
+ "phase": "full_vq",
2247
+ "val_recon_loss": 0.21441739713877775,
2248
+ "val_utilization": 0.951416015625,
2249
+ "time_sec": 136.92144107818604
2250
+ },
2251
+ {
2252
+ "epoch": 173,
2253
+ "loss": 0.2640582248366622,
2254
+ "recon_loss": 0.21103781429409657,
2255
+ "commit_loss": 0.0265102052132329,
2256
+ "embed_loss": 0.0265102052132329,
2257
+ "codebook_utilization": 0.962158203125,
2258
+ "lr": 4.322727117869949e-06,
2259
+ "phase": "full_vq",
2260
+ "val_recon_loss": 0.21443711321675377,
2261
+ "val_utilization": 0.951904296875,
2262
+ "time_sec": 138.40561699867249
2263
+ },
2264
+ {
2265
+ "epoch": 174,
2266
+ "loss": 0.26395986186779613,
2267
+ "recon_loss": 0.21105072328310995,
2268
+ "commit_loss": 0.026454569269627897,
2269
+ "embed_loss": 0.026454569269627897,
2270
+ "codebook_utilization": 0.962158203125,
2271
+ "lr": 4.0010278170587865e-06,
2272
+ "phase": "full_vq",
2273
+ "val_recon_loss": 0.21437096712964304,
2274
+ "val_utilization": 0.953369140625,
2275
+ "time_sec": 137.12302565574646
2276
+ },
2277
+ {
2278
+ "epoch": 175,
2279
+ "loss": 0.26404903405080965,
2280
+ "recon_loss": 0.21105355529119652,
2281
+ "commit_loss": 0.026497739415141303,
2282
+ "embed_loss": 0.026497739415141303,
2283
+ "codebook_utilization": 0.962158203125,
2284
+ "lr": 3.6912675521111804e-06,
2285
+ "phase": "full_vq",
2286
+ "val_recon_loss": 0.2144219952352931,
2287
+ "val_utilization": 0.9521484375,
2288
+ "time_sec": 133.26996755599976
2289
+ },
2290
+ {
2291
+ "epoch": 176,
2292
+ "loss": 0.26405372218226353,
2293
+ "recon_loss": 0.21103870622349302,
2294
+ "commit_loss": 0.026507508004624345,
2295
+ "embed_loss": 0.026507508004624345,
2296
+ "codebook_utilization": 0.9619140625,
2297
+ "lr": 3.393526721321614e-06,
2298
+ "phase": "full_vq",
2299
+ "val_recon_loss": 0.21437660336829303,
2300
+ "val_utilization": 0.951904296875,
2301
+ "time_sec": 138.95040082931519
2302
+ },
2303
+ {
2304
+ "epoch": 177,
2305
+ "loss": 0.26400244332910555,
2306
+ "recon_loss": 0.21097883792104435,
2307
+ "commit_loss": 0.026511802610645934,
2308
+ "embed_loss": 0.026511802610645934,
2309
+ "codebook_utilization": 0.962158203125,
2310
+ "lr": 3.1078826033397826e-06,
2311
+ "phase": "full_vq",
2312
+ "val_recon_loss": 0.2143909346856428,
2313
+ "val_utilization": 0.9521484375,
2314
+ "time_sec": 141.3791856765747
2315
+ },
2316
+ {
2317
+ "epoch": 178,
2318
+ "loss": 0.2640966063107901,
2319
+ "recon_loss": 0.21100135266700087,
2320
+ "commit_loss": 0.02654762703642693,
2321
+ "embed_loss": 0.02654762703642693,
2322
+ "codebook_utilization": 0.962158203125,
2323
+ "lr": 2.834409337112846e-06,
2324
+ "phase": "full_vq",
2325
+ "val_recon_loss": 0.21437338647547732,
2326
+ "val_utilization": 0.9521484375,
2327
+ "time_sec": 145.86063694953918
2328
+ },
2329
+ {
2330
+ "epoch": 179,
2331
+ "loss": 0.2641781853829942,
2332
+ "recon_loss": 0.21105259551911495,
2333
+ "commit_loss": 0.026562794846126704,
2334
+ "embed_loss": 0.026562794846126704,
2335
+ "codebook_utilization": 0.962158203125,
2336
+ "lr": 2.57317790264273e-06,
2337
+ "phase": "full_vq",
2338
+ "val_recon_loss": 0.21438502570551432,
2339
+ "val_utilization": 0.9521484375,
2340
+ "time_sec": 145.466144323349
2341
+ },
2342
+ {
2343
+ "epoch": 180,
2344
+ "loss": 0.2640849213168873,
2345
+ "recon_loss": 0.21105245450324805,
2346
+ "commit_loss": 0.02651623326043288,
2347
+ "embed_loss": 0.02651623326043288,
2348
+ "codebook_utilization": 0.962158203125,
2349
+ "lr": 2.324256102563187e-06,
2350
+ "phase": "full_vq",
2351
+ "val_recon_loss": 0.21437762543726502,
2352
+ "val_utilization": 0.952392578125,
2353
+ "time_sec": 146.26118278503418
2354
+ },
2355
+ {
2356
+ "epoch": 181,
2357
+ "loss": 0.2641721400142039,
2358
+ "recon_loss": 0.21105877944043658,
2359
+ "commit_loss": 0.026556680448413864,
2360
+ "embed_loss": 0.026556680448413864,
2361
+ "codebook_utilization": 0.962158203125,
2362
+ "lr": 2.0877085445416877e-06,
2363
+ "phase": "full_vq",
2364
+ "val_recon_loss": 0.21437230948986633,
2365
+ "val_utilization": 0.95263671875,
2366
+ "time_sec": 137.36411571502686
2367
+ },
2368
+ {
2369
+ "epoch": 182,
2370
+ "loss": 0.2641438046283515,
2371
+ "recon_loss": 0.21100592122572223,
2372
+ "commit_loss": 0.026568941925942575,
2373
+ "embed_loss": 0.026568941925942575,
2374
+ "codebook_utilization": 0.962158203125,
2375
+ "lr": 1.8635966245104596e-06,
2376
+ "phase": "full_vq",
2377
+ "val_recon_loss": 0.21438697089305084,
2378
+ "val_utilization": 0.952880859375,
2379
+ "time_sec": 137.8035717010498
2380
+ },
2381
+ {
2382
+ "epoch": 183,
2383
+ "loss": 0.26420318121267206,
2384
+ "recon_loss": 0.21105521194495483,
2385
+ "commit_loss": 0.026573984515234867,
2386
+ "embed_loss": 0.026573984515234867,
2387
+ "codebook_utilization": 0.962158203125,
2388
+ "lr": 1.651978510731188e-06,
2389
+ "phase": "full_vq",
2390
+ "val_recon_loss": 0.21436885099732475,
2391
+ "val_utilization": 0.952392578125,
2392
+ "time_sec": 139.89997386932373
2393
+ },
2394
+ {
2395
+ "epoch": 184,
2396
+ "loss": 0.26416658248681685,
2397
+ "recon_loss": 0.21100971419717562,
2398
+ "commit_loss": 0.026578434187727085,
2399
+ "embed_loss": 0.026578434187727085,
2400
+ "codebook_utilization": 0.962158203125,
2401
+ "lr": 1.4529091286973986e-06,
2402
+ "phase": "full_vq",
2403
+ "val_recon_loss": 0.21437526033835463,
2404
+ "val_utilization": 0.952880859375,
2405
+ "time_sec": 134.70737767219543
2406
+ },
2407
+ {
2408
+ "epoch": 185,
2409
+ "loss": 0.264119908476102,
2410
+ "recon_loss": 0.21097335653495658,
2411
+ "commit_loss": 0.026573275902427147,
2412
+ "embed_loss": 0.026573275902427147,
2413
+ "codebook_utilization": 0.962158203125,
2414
+ "lr": 1.2664401468786106e-06,
2415
+ "phase": "full_vq",
2416
+ "val_recon_loss": 0.21437027916479645,
2417
+ "val_utilization": 0.953125,
2418
+ "time_sec": 141.36850476264954
2419
+ },
2420
+ {
2421
+ "epoch": 186,
2422
+ "loss": 0.2641745704985893,
2423
+ "recon_loss": 0.21099333685184235,
2424
+ "commit_loss": 0.026590616853660362,
2425
+ "embed_loss": 0.026590616853660362,
2426
+ "codebook_utilization": 0.962158203125,
2427
+ "lr": 1.0926199633097149e-06,
2428
+ "phase": "full_vq",
2429
+ "val_recon_loss": 0.2143677823831526,
2430
+ "val_utilization": 0.953125,
2431
+ "time_sec": 137.4411108493805
2432
+ },
2433
+ {
2434
+ "epoch": 187,
2435
+ "loss": 0.2642669931740619,
2436
+ "recon_loss": 0.2110073836192206,
2437
+ "commit_loss": 0.02662980457803179,
2438
+ "embed_loss": 0.02662980457803179,
2439
+ "codebook_utilization": 0.962158203125,
2440
+ "lr": 9.314936930293277e-07,
2441
+ "phase": "full_vq",
2442
+ "val_recon_loss": 0.21435275129722745,
2443
+ "val_utilization": 0.952392578125,
2444
+ "time_sec": 139.66765689849854
2445
+ },
2446
+ {
2447
+ "epoch": 188,
2448
+ "loss": 0.2641916476129516,
2449
+ "recon_loss": 0.21097810891504856,
2450
+ "commit_loss": 0.026606769555912107,
2451
+ "embed_loss": 0.026606769555912107,
2452
+ "codebook_utilization": 0.962158203125,
2453
+ "lr": 7.83103156370118e-07,
2454
+ "phase": "full_vq",
2455
+ "val_recon_loss": 0.21434208047523928,
2456
+ "val_utilization": 0.9521484375,
2457
+ "time_sec": 135.75908207893372
2458
+ },
2459
+ {
2460
+ "epoch": 189,
2461
+ "loss": 0.26420647703454425,
2462
+ "recon_loss": 0.2110019476836936,
2463
+ "commit_loss": 0.026602264435653927,
2464
+ "embed_loss": 0.026602264435653927,
2465
+ "codebook_utilization": 0.962158203125,
2466
+ "lr": 6.474868681043573e-07,
2467
+ "phase": "full_vq",
2468
+ "val_recon_loss": 0.21436412632465363,
2469
+ "val_utilization": 0.952392578125,
2470
+ "time_sec": 138.20861744880676
2471
+ },
2472
+ {
2473
+ "epoch": 190,
2474
+ "loss": 0.26420683474317797,
2475
+ "recon_loss": 0.21098680875004144,
2476
+ "commit_loss": 0.026610013185861474,
2477
+ "embed_loss": 0.026610013185861474,
2478
+ "codebook_utilization": 0.962158203125,
2479
+ "lr": 5.246800274474491e-07,
2480
+ "phase": "full_vq",
2481
+ "val_recon_loss": 0.2143944591451227,
2482
+ "val_utilization": 0.952880859375,
2483
+ "time_sec": 136.80305171012878
2484
+ },
2485
+ {
2486
+ "epoch": 191,
2487
+ "loss": 0.26419439625126234,
2488
+ "recon_loss": 0.21100580534799313,
2489
+ "commit_loss": 0.026594295345630427,
2490
+ "embed_loss": 0.026594295345630427,
2491
+ "codebook_utilization": 0.962158203125,
2492
+ "lr": 4.1471450892189815e-07,
2493
+ "phase": "full_vq",
2494
+ "val_recon_loss": 0.21437959649254767,
2495
+ "val_utilization": 0.953125,
2496
+ "time_sec": 142.38911080360413
2497
+ },
2498
+ {
2499
+ "epoch": 192,
2500
+ "loss": 0.26418292526468673,
2501
+ "recon_loss": 0.21096790058987575,
2502
+ "commit_loss": 0.026607512420694517,
2503
+ "embed_loss": 0.026607512420694517,
2504
+ "codebook_utilization": 0.962158203125,
2505
+ "lr": 3.1761885408435033e-07,
2506
+ "phase": "full_vq",
2507
+ "val_recon_loss": 0.21437878638840793,
2508
+ "val_utilization": 0.95263671875,
2509
+ "time_sec": 134.47354292869568
2510
+ },
2511
+ {
2512
+ "epoch": 193,
2513
+ "loss": 0.26414327565732043,
2514
+ "recon_loss": 0.21094106317454883,
2515
+ "commit_loss": 0.026601106074807767,
2516
+ "embed_loss": 0.026601106074807767,
2517
+ "codebook_utilization": 0.962158203125,
2518
+ "lr": 2.3341826411756845e-07,
2519
+ "phase": "full_vq",
2520
+ "val_recon_loss": 0.21436425323566694,
2521
+ "val_utilization": 0.953125,
2522
+ "time_sec": 132.6182668209076
2523
+ },
2524
+ {
2525
+ "epoch": 194,
2526
+ "loss": 0.2641559117006739,
2527
+ "recon_loss": 0.210946664638151,
2528
+ "commit_loss": 0.026604623614550446,
2529
+ "embed_loss": 0.026604623614550446,
2530
+ "codebook_utilization": 0.962158203125,
2531
+ "lr": 1.6213459328950342e-07,
2532
+ "phase": "full_vq",
2533
+ "val_recon_loss": 0.21434766096010638,
2534
+ "val_utilization": 0.952880859375,
2535
+ "time_sec": 139.21892857551575
2536
+ },
2537
+ {
2538
+ "epoch": 195,
2539
+ "loss": 0.26417293405629755,
2540
+ "recon_loss": 0.21093663482404337,
2541
+ "commit_loss": 0.026618149704463922,
2542
+ "embed_loss": 0.026618149704463922,
2543
+ "codebook_utilization": 0.962158203125,
2544
+ "lr": 1.037863432809926e-07,
2545
+ "phase": "full_vq",
2546
+ "val_recon_loss": 0.21434564453162505,
2547
+ "val_utilization": 0.952880859375,
2548
+ "time_sec": 136.1987156867981
2549
+ },
2550
+ {
2551
+ "epoch": 196,
2552
+ "loss": 0.26416726018678205,
2553
+ "recon_loss": 0.21095861292628415,
2554
+ "commit_loss": 0.02660432383973343,
2555
+ "embed_loss": 0.02660432383973343,
2556
+ "codebook_utilization": 0.962158203125,
2557
+ "lr": 5.8388658383673425e-08,
2558
+ "phase": "full_vq",
2559
+ "val_recon_loss": 0.21434264420793298,
2560
+ "val_utilization": 0.952880859375,
2561
+ "time_sec": 135.9748466014862
2562
+ },
2563
+ {
2564
+ "epoch": 197,
2565
+ "loss": 0.2641810091126579,
2566
+ "recon_loss": 0.2109540049945759,
2567
+ "commit_loss": 0.026613502165045197,
2568
+ "embed_loss": 0.026613502165045197,
2569
+ "codebook_utilization": 0.962158203125,
2570
+ "lr": 2.5953321569255322e-08,
2571
+ "phase": "full_vq",
2572
+ "val_recon_loss": 0.2143414467573166,
2573
+ "val_utilization": 0.952880859375,
2574
+ "time_sec": 134.14156460762024
2575
+ },
2576
+ {
2577
+ "epoch": 198,
2578
+ "loss": 0.26419750897864985,
2579
+ "recon_loss": 0.21097038756944947,
2580
+ "commit_loss": 0.026613560628882916,
2581
+ "embed_loss": 0.026613560628882916,
2582
+ "codebook_utilization": 0.962158203125,
2583
+ "lr": 6.4887514312661454e-09,
2584
+ "phase": "full_vq",
2585
+ "val_recon_loss": 0.21434263014391566,
2586
+ "val_utilization": 0.952880859375,
2587
+ "time_sec": 142.8901333808899
2588
+ },
2589
+ {
2590
+ "epoch": 199,
2591
+ "loss": 0.2642024381939312,
2592
+ "recon_loss": 0.2109852933342541,
2593
+ "commit_loss": 0.0266085722582127,
2594
+ "embed_loss": 0.0266085722582127,
2595
+ "codebook_utilization": 0.962158203125,
2596
+ "lr": 0.0,
2597
+ "phase": "full_vq",
2598
+ "val_recon_loss": 0.21434488147497177,
2599
+ "val_utilization": 0.952880859375,
2600
+ "time_sec": 148.42288064956665
2601
+ }
2602
+ ]