Pthahnix commited on
Commit
ec34246
·
verified ·
1 Parent(s): 38090d3

Upload checkpoints/exp1_A_5cat/training_history.json with huggingface_hub

Browse files
checkpoints/exp1_A_5cat/training_history.json ADDED
@@ -0,0 +1,2602 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "epoch": 0,
4
+ "loss": 0.6486299189866758,
5
+ "recon_loss": 0.6486299189866758,
6
+ "commit_loss": 0.28864870760955064,
7
+ "embed_loss": 0.28864870760955064,
8
+ "codebook_utilization": 0.007080078125,
9
+ "lr": 2.08e-05,
10
+ "phase": "encoder_warmup",
11
+ "val_recon_loss": 0.6882578868132371,
12
+ "val_utilization": 0.00634765625,
13
+ "time_sec": 49.90796446800232
14
+ },
15
+ {
16
+ "epoch": 1,
17
+ "loss": 0.5728353998240303,
18
+ "recon_loss": 0.5728353998240303,
19
+ "commit_loss": 0.3421454038105759,
20
+ "embed_loss": 0.3421454038105759,
21
+ "codebook_utilization": 0.01171875,
22
+ "lr": 4.06e-05,
23
+ "phase": "encoder_warmup",
24
+ "val_recon_loss": 0.5380583176246057,
25
+ "val_utilization": 0.00439453125,
26
+ "time_sec": 47.82359480857849
27
+ },
28
+ {
29
+ "epoch": 2,
30
+ "loss": 0.42713443496648,
31
+ "recon_loss": 0.42713443496648,
32
+ "commit_loss": 0.4958566392169279,
33
+ "embed_loss": 0.4958566392169279,
34
+ "codebook_utilization": 0.0107421875,
35
+ "lr": 6.04e-05,
36
+ "phase": "encoder_warmup",
37
+ "val_recon_loss": 0.4003955928178934,
38
+ "val_utilization": 0.0068359375,
39
+ "time_sec": 46.994564056396484
40
+ },
41
+ {
42
+ "epoch": 3,
43
+ "loss": 0.36442408316275654,
44
+ "recon_loss": 0.36442408316275654,
45
+ "commit_loss": 0.46526922022595124,
46
+ "embed_loss": 0.46526922022595124,
47
+ "codebook_utilization": 0.01025390625,
48
+ "lr": 8.02e-05,
49
+ "phase": "encoder_warmup",
50
+ "val_recon_loss": 0.37920735203302824,
51
+ "val_utilization": 0.0078125,
52
+ "time_sec": 47.1387722492218
53
+ },
54
+ {
55
+ "epoch": 4,
56
+ "loss": 0.3522573055005541,
57
+ "recon_loss": 0.3522573055005541,
58
+ "commit_loss": 0.334042723272361,
59
+ "embed_loss": 0.334042723272361,
60
+ "codebook_utilization": 0.02001953125,
61
+ "lr": 0.0001,
62
+ "phase": "encoder_warmup",
63
+ "val_recon_loss": 0.3648387079055493,
64
+ "val_utilization": 0.01220703125,
65
+ "time_sec": 47.076200008392334
66
+ },
67
+ {
68
+ "epoch": 5,
69
+ "loss": 0.3447116917254878,
70
+ "recon_loss": 0.3447116917254878,
71
+ "commit_loss": 0.33499857432702007,
72
+ "embed_loss": 0.33499857432702007,
73
+ "codebook_utilization": 0.020751953125,
74
+ "lr": 9.999351124856874e-05,
75
+ "phase": "encoder_warmup",
76
+ "val_recon_loss": 0.35950949558844936,
77
+ "val_utilization": 0.009521484375,
78
+ "time_sec": 47.2284939289093
79
+ },
80
+ {
81
+ "epoch": 6,
82
+ "loss": 0.3363506700478348,
83
+ "recon_loss": 0.3363506700478348,
84
+ "commit_loss": 0.3566281655255486,
85
+ "embed_loss": 0.3566281655255486,
86
+ "codebook_utilization": 0.01513671875,
87
+ "lr": 9.997404667843075e-05,
88
+ "phase": "encoder_warmup",
89
+ "val_recon_loss": 0.3472484075106107,
90
+ "val_utilization": 0.0087890625,
91
+ "time_sec": 47.250046491622925
92
+ },
93
+ {
94
+ "epoch": 7,
95
+ "loss": 0.3301704222080754,
96
+ "recon_loss": 0.3301704222080754,
97
+ "commit_loss": 0.36232977521185783,
98
+ "embed_loss": 0.36232977521185783,
99
+ "codebook_utilization": 0.015380859375,
100
+ "lr": 9.994161134161632e-05,
101
+ "phase": "encoder_warmup",
102
+ "val_recon_loss": 0.3396035639139322,
103
+ "val_utilization": 0.009765625,
104
+ "time_sec": 46.06647348403931
105
+ },
106
+ {
107
+ "epoch": 8,
108
+ "loss": 0.32530146895670425,
109
+ "recon_loss": 0.32530146895670425,
110
+ "commit_loss": 0.34497708900302065,
111
+ "embed_loss": 0.34497708900302065,
112
+ "codebook_utilization": 0.017822265625,
113
+ "lr": 9.9896213656719e-05,
114
+ "phase": "encoder_warmup",
115
+ "val_recon_loss": 0.33421732829167294,
116
+ "val_utilization": 0.010498046875,
117
+ "time_sec": 47.76232647895813
118
+ },
119
+ {
120
+ "epoch": 9,
121
+ "loss": 0.3226450932960884,
122
+ "recon_loss": 0.3226450932960884,
123
+ "commit_loss": 0.34663038335594476,
124
+ "embed_loss": 0.34663038335594476,
125
+ "codebook_utilization": 0.018310546875,
126
+ "lr": 9.98378654067105e-05,
127
+ "phase": "encoder_warmup",
128
+ "val_recon_loss": 0.7123874104939975,
129
+ "val_utilization": 0.183349609375,
130
+ "time_sec": 67.86261558532715
131
+ },
132
+ {
133
+ "epoch": 10,
134
+ "loss": 0.4732203243994245,
135
+ "recon_loss": 0.4405503424943662,
136
+ "commit_loss": 0.01633499261430081,
137
+ "embed_loss": 0.01633499261430081,
138
+ "codebook_utilization": 0.2998046875,
139
+ "lr": 9.976658173588243e-05,
140
+ "phase": "full_vq",
141
+ "val_recon_loss": 0.3527405009819911,
142
+ "val_utilization": 0.075927734375,
143
+ "time_sec": 48.49461889266968
144
+ },
145
+ {
146
+ "epoch": 11,
147
+ "loss": 0.34965790486803244,
148
+ "recon_loss": 0.32759416278670817,
149
+ "commit_loss": 0.011031871095445811,
150
+ "embed_loss": 0.011031871095445811,
151
+ "codebook_utilization": 0.17578125,
152
+ "lr": 9.968238114591565e-05,
153
+ "phase": "full_vq",
154
+ "val_recon_loss": 0.33133659683741057,
155
+ "val_utilization": 0.123291015625,
156
+ "time_sec": 47.27166962623596
157
+ },
158
+ {
159
+ "epoch": 12,
160
+ "loss": 0.3373577361013375,
161
+ "recon_loss": 0.3142113744043836,
162
+ "commit_loss": 0.011573182163285274,
163
+ "embed_loss": 0.011573182163285274,
164
+ "codebook_utilization": 0.225341796875,
165
+ "lr": 9.95852854910781e-05,
166
+ "phase": "full_vq",
167
+ "val_recon_loss": 0.3183709245461684,
168
+ "val_utilization": 0.135986328125,
169
+ "time_sec": 46.27233028411865
170
+ },
171
+ {
172
+ "epoch": 13,
173
+ "loss": 0.3316730745867187,
174
+ "recon_loss": 0.3073914915907617,
175
+ "commit_loss": 0.012140789671855815,
176
+ "embed_loss": 0.012140789671855815,
177
+ "codebook_utilization": 0.25,
178
+ "lr": 9.947531997255256e-05,
179
+ "phase": "full_vq",
180
+ "val_recon_loss": 0.31398263344397914,
181
+ "val_utilization": 0.147216796875,
182
+ "time_sec": 47.78503632545471
183
+ },
184
+ {
185
+ "epoch": 14,
186
+ "loss": 0.3248127891736872,
187
+ "recon_loss": 0.30073029562538744,
188
+ "commit_loss": 0.012041248435921529,
189
+ "embed_loss": 0.012041248435921529,
190
+ "codebook_utilization": 0.26806640625,
191
+ "lr": 9.935251313189564e-05,
192
+ "phase": "full_vq",
193
+ "val_recon_loss": 0.30578744411468506,
194
+ "val_utilization": 0.1533203125,
195
+ "time_sec": 47.05907416343689
196
+ },
197
+ {
198
+ "epoch": 15,
199
+ "loss": 0.3204253257489672,
200
+ "recon_loss": 0.29635289954204186,
201
+ "commit_loss": 0.012036213048678987,
202
+ "embed_loss": 0.012036213048678987,
203
+ "codebook_utilization": 0.270263671875,
204
+ "lr": 9.921689684362989e-05,
205
+ "phase": "full_vq",
206
+ "val_recon_loss": 0.3024442814863645,
207
+ "val_utilization": 0.156494140625,
208
+ "time_sec": 47.72433543205261
209
+ },
210
+ {
211
+ "epoch": 16,
212
+ "loss": 0.3182387801946378,
213
+ "recon_loss": 0.29361236329172175,
214
+ "commit_loss": 0.012313208086233512,
215
+ "embed_loss": 0.012313208086233512,
216
+ "codebook_utilization": 0.287353515625,
217
+ "lr": 9.906850630697068e-05,
218
+ "phase": "full_vq",
219
+ "val_recon_loss": 0.3008571748550122,
220
+ "val_utilization": 0.162841796875,
221
+ "time_sec": 46.76892352104187
222
+ },
223
+ {
224
+ "epoch": 17,
225
+ "loss": 0.31541585221010093,
226
+ "recon_loss": 0.2906547147853702,
227
+ "commit_loss": 0.012380570556749316,
228
+ "embed_loss": 0.012380570556749316,
229
+ "codebook_utilization": 0.295166015625,
230
+ "lr": 9.890738003669029e-05,
231
+ "phase": "full_vq",
232
+ "val_recon_loss": 0.29535558361273545,
233
+ "val_utilization": 0.1767578125,
234
+ "time_sec": 47.050782918930054
235
+ },
236
+ {
237
+ "epoch": 18,
238
+ "loss": 0.3123515300890979,
239
+ "recon_loss": 0.28801636602364333,
240
+ "commit_loss": 0.012167582562302841,
241
+ "embed_loss": 0.012167582562302841,
242
+ "codebook_utilization": 0.301513671875,
243
+ "lr": 9.87335598531214e-05,
244
+ "phase": "full_vq",
245
+ "val_recon_loss": 0.2925591789759122,
246
+ "val_utilization": 0.18994140625,
247
+ "time_sec": 48.1220269203186
248
+ },
249
+ {
250
+ "epoch": 19,
251
+ "loss": 0.3114185701398289,
252
+ "recon_loss": 0.28645869855787237,
253
+ "commit_loss": 0.012479938804080673,
254
+ "embed_loss": 0.012479938804080673,
255
+ "codebook_utilization": 0.32958984375,
256
+ "lr": 9.85470908713026e-05,
257
+ "phase": "full_vq",
258
+ "val_recon_loss": 0.2966208549646231,
259
+ "val_utilization": 0.185791015625,
260
+ "time_sec": 47.16935706138611
261
+ },
262
+ {
263
+ "epoch": 20,
264
+ "loss": 0.31054385444697213,
265
+ "recon_loss": 0.286402121478436,
266
+ "commit_loss": 0.012070866027737366,
267
+ "embed_loss": 0.012070866027737366,
268
+ "codebook_utilization": 0.309326171875,
269
+ "lr": 9.834802148926882e-05,
270
+ "phase": "full_vq",
271
+ "val_recon_loss": 0.2922164774858035,
272
+ "val_utilization": 0.180908203125,
273
+ "time_sec": 47.45058584213257
274
+ },
275
+ {
276
+ "epoch": 21,
277
+ "loss": 0.3073745632872862,
278
+ "recon_loss": 0.2834460717790267,
279
+ "commit_loss": 0.011964245881958335,
280
+ "embed_loss": 0.011964245881958335,
281
+ "codebook_utilization": 0.300537109375,
282
+ "lr": 9.813640337548953e-05,
283
+ "phase": "full_vq",
284
+ "val_recon_loss": 0.28753021703316617,
285
+ "val_utilization": 0.18896484375,
286
+ "time_sec": 47.10411715507507
287
+ },
288
+ {
289
+ "epoch": 22,
290
+ "loss": 0.30539435498854695,
291
+ "recon_loss": 0.2813229823813719,
292
+ "commit_loss": 0.012035686723595741,
293
+ "embed_loss": 0.012035686723595741,
294
+ "codebook_utilization": 0.294677734375,
295
+ "lr": 9.79122914554583e-05,
296
+ "phase": "full_vq",
297
+ "val_recon_loss": 0.28625730597055876,
298
+ "val_utilization": 0.186767578125,
299
+ "time_sec": 46.970909118652344
300
+ },
301
+ {
302
+ "epoch": 23,
303
+ "loss": 0.3050253987312317,
304
+ "recon_loss": 0.28003353231093464,
305
+ "commit_loss": 0.01249593363015675,
306
+ "embed_loss": 0.01249593363015675,
307
+ "codebook_utilization": 0.2958984375,
308
+ "lr": 9.76757438974368e-05,
309
+ "phase": "full_vq",
310
+ "val_recon_loss": 0.28434716967435986,
311
+ "val_utilization": 0.18798828125,
312
+ "time_sec": 46.632248401641846
313
+ },
314
+ {
315
+ "epoch": 24,
316
+ "loss": 0.301952743062786,
317
+ "recon_loss": 0.2778947213116814,
318
+ "commit_loss": 0.012029010948597216,
319
+ "embed_loss": 0.012029010948597216,
320
+ "codebook_utilization": 0.288818359375,
321
+ "lr": 9.742682209735726e-05,
322
+ "phase": "full_vq",
323
+ "val_recon_loss": 0.28360043695339787,
324
+ "val_utilization": 0.185546875,
325
+ "time_sec": 47.21303033828735
326
+ },
327
+ {
328
+ "epoch": 25,
329
+ "loss": 0.3012464426311792,
330
+ "recon_loss": 0.27732254652416005,
331
+ "commit_loss": 0.011961946683917559,
332
+ "embed_loss": 0.011961946683917559,
333
+ "codebook_utilization": 0.286865234375,
334
+ "lr": 9.716559066288714e-05,
335
+ "phase": "full_vq",
336
+ "val_recon_loss": 0.2811767275516803,
337
+ "val_utilization": 0.191162109375,
338
+ "time_sec": 48.00208783149719
339
+ },
340
+ {
341
+ "epoch": 26,
342
+ "loss": 0.3003270702034819,
343
+ "recon_loss": 0.2757678154636832,
344
+ "commit_loss": 0.012279627388160602,
345
+ "embed_loss": 0.012279627388160602,
346
+ "codebook_utilization": 0.287841796875,
347
+ "lr": 9.689211739666022e-05,
348
+ "phase": "full_vq",
349
+ "val_recon_loss": 0.280276188483605,
350
+ "val_utilization": 0.19384765625,
351
+ "time_sec": 48.07919430732727
352
+ },
353
+ {
354
+ "epoch": 27,
355
+ "loss": 0.3004833255328384,
356
+ "recon_loss": 0.27466156844999273,
357
+ "commit_loss": 0.012910877062263442,
358
+ "embed_loss": 0.012910877062263442,
359
+ "codebook_utilization": 0.284912109375,
360
+ "lr": 9.660647327867839e-05,
361
+ "phase": "full_vq",
362
+ "val_recon_loss": 0.2790587819539584,
363
+ "val_utilization": 0.189208984375,
364
+ "time_sec": 48.270617723464966
365
+ },
366
+ {
367
+ "epoch": 28,
368
+ "loss": 0.297929081846686,
369
+ "recon_loss": 0.2733270870704277,
370
+ "commit_loss": 0.012300998246406807,
371
+ "embed_loss": 0.012300998246406807,
372
+ "codebook_utilization": 0.284912109375,
373
+ "lr": 9.630873244788881e-05,
374
+ "phase": "full_vq",
375
+ "val_recon_loss": 0.2779857344352282,
376
+ "val_utilization": 0.193359375,
377
+ "time_sec": 48.9530029296875
378
+ },
379
+ {
380
+ "epoch": 29,
381
+ "loss": 0.29830770340620305,
382
+ "recon_loss": 0.27251628800934435,
383
+ "commit_loss": 0.012895706821890437,
384
+ "embed_loss": 0.012895706821890437,
385
+ "codebook_utilization": 0.287109375,
386
+ "lr": 9.59989721829412e-05,
387
+ "phase": "full_vq",
388
+ "val_recon_loss": 0.27671571992910826,
389
+ "val_utilization": 0.192626953125,
390
+ "time_sec": 45.41763091087341
391
+ },
392
+ {
393
+ "epoch": 30,
394
+ "loss": 0.298766399715461,
395
+ "recon_loss": 0.27223750598290386,
396
+ "commit_loss": 0.013264447030629598,
397
+ "embed_loss": 0.013264447030629598,
398
+ "codebook_utilization": 0.275634765625,
399
+ "lr": 9.567727288213003e-05,
400
+ "phase": "full_vq",
401
+ "val_recon_loss": 0.27744454603928786,
402
+ "val_utilization": 0.193115234375,
403
+ "time_sec": 45.646090507507324
404
+ },
405
+ {
406
+ "epoch": 31,
407
+ "loss": 0.2983699201368818,
408
+ "recon_loss": 0.27142763546868864,
409
+ "commit_loss": 0.013471142188006756,
410
+ "embed_loss": 0.013471142188006756,
411
+ "codebook_utilization": 0.276123046875,
412
+ "lr": 9.534371804252726e-05,
413
+ "phase": "full_vq",
414
+ "val_recon_loss": 0.2765607214890994,
415
+ "val_utilization": 0.1923828125,
416
+ "time_sec": 47.64370608329773
417
+ },
418
+ {
419
+ "epoch": 32,
420
+ "loss": 0.29894952154627036,
421
+ "recon_loss": 0.2710699737071991,
422
+ "commit_loss": 0.01393977205689047,
423
+ "embed_loss": 0.01393977205689047,
424
+ "codebook_utilization": 0.27294921875,
425
+ "lr": 9.499839423831059e-05,
426
+ "phase": "full_vq",
427
+ "val_recon_loss": 0.27488876306093657,
428
+ "val_utilization": 0.189208984375,
429
+ "time_sec": 47.62825846672058
430
+ },
431
+ {
432
+ "epoch": 33,
433
+ "loss": 0.2972109463869357,
434
+ "recon_loss": 0.2695390535335915,
435
+ "commit_loss": 0.013835947595390618,
436
+ "embed_loss": 0.013835947595390618,
437
+ "codebook_utilization": 0.27099609375,
438
+ "lr": 9.464139109829319e-05,
439
+ "phase": "full_vq",
440
+ "val_recon_loss": 0.27520145361240095,
441
+ "val_utilization": 0.189453125,
442
+ "time_sec": 46.83393621444702
443
+ },
444
+ {
445
+ "epoch": 34,
446
+ "loss": 0.2966874884624107,
447
+ "recon_loss": 0.2687463222765455,
448
+ "commit_loss": 0.013970581778124267,
449
+ "embed_loss": 0.013970581778124267,
450
+ "codebook_utilization": 0.272705078125,
451
+ "lr": 9.427280128266047e-05,
452
+ "phase": "full_vq",
453
+ "val_recon_loss": 0.27270328540068406,
454
+ "val_utilization": 0.187744140625,
455
+ "time_sec": 47.61192607879639
456
+ },
457
+ {
458
+ "epoch": 35,
459
+ "loss": 0.29720970754529913,
460
+ "recon_loss": 0.2682199302841635,
461
+ "commit_loss": 0.014494888447955543,
462
+ "embed_loss": 0.014494888447955543,
463
+ "codebook_utilization": 0.26904296875,
464
+ "lr": 9.389272045892021e-05,
465
+ "phase": "full_vq",
466
+ "val_recon_loss": 0.2726908910733003,
467
+ "val_utilization": 0.194091796875,
468
+ "time_sec": 48.51250982284546
469
+ },
470
+ {
471
+ "epoch": 36,
472
+ "loss": 0.29454753504079934,
473
+ "recon_loss": 0.2666793617547727,
474
+ "commit_loss": 0.013934086204743852,
475
+ "embed_loss": 0.013934086204743852,
476
+ "codebook_utilization": 0.269287109375,
477
+ "lr": 9.350124727707194e-05,
478
+ "phase": "full_vq",
479
+ "val_recon_loss": 0.27167830100426305,
480
+ "val_utilization": 0.194091796875,
481
+ "time_sec": 47.0704140663147
482
+ },
483
+ {
484
+ "epoch": 37,
485
+ "loss": 0.29581303164070727,
486
+ "recon_loss": 0.26675746984341564,
487
+ "commit_loss": 0.014527780697772316,
488
+ "embed_loss": 0.014527780697772316,
489
+ "codebook_utilization": 0.2685546875,
490
+ "lr": 9.309848334400244e-05,
491
+ "phase": "full_vq",
492
+ "val_recon_loss": 0.27314719213889194,
493
+ "val_utilization": 0.19482421875,
494
+ "time_sec": 46.51348829269409
495
+ },
496
+ {
497
+ "epoch": 38,
498
+ "loss": 0.29552772173694536,
499
+ "recon_loss": 0.2656755716192956,
500
+ "commit_loss": 0.014926076665812847,
501
+ "embed_loss": 0.014926076665812847,
502
+ "codebook_utilization": 0.270263671875,
503
+ "lr": 9.26845331971136e-05,
504
+ "phase": "full_vq",
505
+ "val_recon_loss": 0.26968666452627915,
506
+ "val_utilization": 0.195556640625,
507
+ "time_sec": 45.83749270439148
508
+ },
509
+ {
510
+ "epoch": 39,
511
+ "loss": 0.29380739553301943,
512
+ "recon_loss": 0.26442454609216426,
513
+ "commit_loss": 0.014691425286525605,
514
+ "embed_loss": 0.014691425286525605,
515
+ "codebook_utilization": 0.268798828125,
516
+ "lr": 9.225950427718973e-05,
517
+ "phase": "full_vq",
518
+ "val_recon_loss": 0.2683155181316229,
519
+ "val_utilization": 0.1962890625,
520
+ "time_sec": 46.41405630111694
521
+ },
522
+ {
523
+ "epoch": 40,
524
+ "loss": 0.29542936998255115,
525
+ "recon_loss": 0.26448582375750823,
526
+ "commit_loss": 0.015471771414227346,
527
+ "embed_loss": 0.015471771414227346,
528
+ "codebook_utilization": 0.25927734375,
529
+ "lr": 9.182350690051132e-05,
530
+ "phase": "full_vq",
531
+ "val_recon_loss": 0.26950591114851147,
532
+ "val_utilization": 0.19580078125,
533
+ "time_sec": 47.56708574295044
534
+ },
535
+ {
536
+ "epoch": 41,
537
+ "loss": 0.29732147911015677,
538
+ "recon_loss": 0.2642808235159107,
539
+ "commit_loss": 0.016520327961474072,
540
+ "embed_loss": 0.016520327961474072,
541
+ "codebook_utilization": 0.260009765625,
542
+ "lr": 9.137665423022247e-05,
543
+ "phase": "full_vq",
544
+ "val_recon_loss": 0.27156637952877927,
545
+ "val_utilization": 0.18603515625,
546
+ "time_sec": 46.73802447319031
547
+ },
548
+ {
549
+ "epoch": 42,
550
+ "loss": 0.2970982661434248,
551
+ "recon_loss": 0.26403798659642536,
552
+ "commit_loss": 0.01653013990132832,
553
+ "embed_loss": 0.01653013990132832,
554
+ "codebook_utilization": 0.260498046875,
555
+ "lr": 9.091906224695934e-05,
556
+ "phase": "full_vq",
557
+ "val_recon_loss": 0.2682831252996738,
558
+ "val_utilization": 0.194580078125,
559
+ "time_sec": 47.08918333053589
560
+ },
561
+ {
562
+ "epoch": 43,
563
+ "loss": 0.29397598493332955,
564
+ "recon_loss": 0.2627227326818541,
565
+ "commit_loss": 0.01562662594312546,
566
+ "embed_loss": 0.01562662594312546,
567
+ "codebook_utilization": 0.258056640625,
568
+ "lr": 9.045084971874735e-05,
569
+ "phase": "full_vq",
570
+ "val_recon_loss": 0.2666227863385127,
571
+ "val_utilization": 0.194580078125,
572
+ "time_sec": 48.25100255012512
573
+ },
574
+ {
575
+ "epoch": 44,
576
+ "loss": 0.2952479375343697,
577
+ "recon_loss": 0.26203199753574297,
578
+ "commit_loss": 0.016607969579305135,
579
+ "embed_loss": 0.016607969579305135,
580
+ "codebook_utilization": 0.25927734375,
581
+ "lr": 8.997213817017505e-05,
582
+ "phase": "full_vq",
583
+ "val_recon_loss": 0.26836622792940873,
584
+ "val_utilization": 0.1865234375,
585
+ "time_sec": 46.84050416946411
586
+ },
587
+ {
588
+ "epoch": 45,
589
+ "loss": 0.29579194971159395,
590
+ "recon_loss": 0.26219205236902426,
591
+ "commit_loss": 0.016799948653023616,
592
+ "embed_loss": 0.016799948653023616,
593
+ "codebook_utilization": 0.258544921875,
594
+ "lr": 8.948305185085224e-05,
595
+ "phase": "full_vq",
596
+ "val_recon_loss": 0.2660675576099983,
597
+ "val_utilization": 0.19873046875,
598
+ "time_sec": 47.09797644615173
599
+ },
600
+ {
601
+ "epoch": 46,
602
+ "loss": 0.29582536571166096,
603
+ "recon_loss": 0.2616012441176994,
604
+ "commit_loss": 0.017112060906548127,
605
+ "embed_loss": 0.017112060906548127,
606
+ "codebook_utilization": 0.2568359375,
607
+ "lr": 8.89837177031611e-05,
608
+ "phase": "full_vq",
609
+ "val_recon_loss": 0.26639335315961105,
610
+ "val_utilization": 0.198486328125,
611
+ "time_sec": 47.34291744232178
612
+ },
613
+ {
614
+ "epoch": 47,
615
+ "loss": 0.2973482030279496,
616
+ "recon_loss": 0.2612737835037942,
617
+ "commit_loss": 0.018037209908167522,
618
+ "embed_loss": 0.018037209908167522,
619
+ "codebook_utilization": 0.25830078125,
620
+ "lr": 8.847426532930828e-05,
621
+ "phase": "full_vq",
622
+ "val_recon_loss": 0.2669075968173834,
623
+ "val_utilization": 0.19384765625,
624
+ "time_sec": 48.65843629837036
625
+ },
626
+ {
627
+ "epoch": 48,
628
+ "loss": 0.2975584671777837,
629
+ "recon_loss": 0.26122236894626244,
630
+ "commit_loss": 0.018168047618340042,
631
+ "embed_loss": 0.018168047618340042,
632
+ "codebook_utilization": 0.254150390625,
633
+ "lr": 8.795482695768655e-05,
634
+ "phase": "full_vq",
635
+ "val_recon_loss": 0.26514462668162125,
636
+ "val_utilization": 0.18310546875,
637
+ "time_sec": 48.06810641288757
638
+ },
639
+ {
640
+ "epoch": 49,
641
+ "loss": 0.29550563412554126,
642
+ "recon_loss": 0.2599199057794085,
643
+ "commit_loss": 0.017792863515662213,
644
+ "embed_loss": 0.017792863515662213,
645
+ "codebook_utilization": 0.253662109375,
646
+ "lr": 8.742553740855503e-05,
647
+ "phase": "full_vq",
648
+ "val_recon_loss": 0.26592662128118366,
649
+ "val_utilization": 0.1865234375,
650
+ "time_sec": 47.01236057281494
651
+ },
652
+ {
653
+ "epoch": 50,
654
+ "loss": 0.29589442470494437,
655
+ "recon_loss": 0.2600334529198852,
656
+ "commit_loss": 0.017930486915158292,
657
+ "embed_loss": 0.017930486915158292,
658
+ "codebook_utilization": 0.248779296875,
659
+ "lr": 8.688653405904649e-05,
660
+ "phase": "full_vq",
661
+ "val_recon_loss": 0.2641307207254263,
662
+ "val_utilization": 0.19287109375,
663
+ "time_sec": 45.94614338874817
664
+ },
665
+ {
666
+ "epoch": 51,
667
+ "loss": 0.2972603101356357,
668
+ "recon_loss": 0.2594982531725192,
669
+ "commit_loss": 0.018881028554603167,
670
+ "embed_loss": 0.018881028554603167,
671
+ "codebook_utilization": 0.247802734375,
672
+ "lr": 8.633795680751113e-05,
673
+ "phase": "full_vq",
674
+ "val_recon_loss": 0.2630243943287776,
675
+ "val_utilization": 0.197509765625,
676
+ "time_sec": 44.92711162567139
677
+ },
678
+ {
679
+ "epoch": 52,
680
+ "loss": 0.296084136355157,
681
+ "recon_loss": 0.2581243921144336,
682
+ "commit_loss": 0.018979872412541333,
683
+ "embed_loss": 0.018979872412541333,
684
+ "codebook_utilization": 0.248291015625,
685
+ "lr": 8.577994803720604e-05,
686
+ "phase": "full_vq",
687
+ "val_recon_loss": 0.2629389247068992,
688
+ "val_utilization": 0.192138671875,
689
+ "time_sec": 44.47393608093262
690
+ },
691
+ {
692
+ "epoch": 53,
693
+ "loss": 0.2957160747518726,
694
+ "recon_loss": 0.257467662879065,
695
+ "commit_loss": 0.01912420626510592,
696
+ "embed_loss": 0.01912420626510592,
697
+ "codebook_utilization": 0.245361328125,
698
+ "lr": 8.521265257933945e-05,
699
+ "phase": "full_vq",
700
+ "val_recon_loss": 0.2618474054795045,
701
+ "val_utilization": 0.1923828125,
702
+ "time_sec": 45.64769721031189
703
+ },
704
+ {
705
+ "epoch": 54,
706
+ "loss": 0.2967555522918701,
707
+ "recon_loss": 0.2571786265162861,
708
+ "commit_loss": 0.019788462011253134,
709
+ "embed_loss": 0.019788462011253134,
710
+ "codebook_utilization": 0.246826171875,
711
+ "lr": 8.463621767547996e-05,
712
+ "phase": "full_vq",
713
+ "val_recon_loss": 0.2621013934795673,
714
+ "val_utilization": 0.1962890625,
715
+ "time_sec": 47.84793281555176
716
+ },
717
+ {
718
+ "epoch": 55,
719
+ "loss": 0.295552766790577,
720
+ "recon_loss": 0.2565985833312951,
721
+ "commit_loss": 0.01947709125484906,
722
+ "embed_loss": 0.01947709125484906,
723
+ "codebook_utilization": 0.2470703125,
724
+ "lr": 8.405079293933984e-05,
725
+ "phase": "full_vq",
726
+ "val_recon_loss": 0.26097967418340534,
727
+ "val_utilization": 0.1962890625,
728
+ "time_sec": 48.29246258735657
729
+ },
730
+ {
731
+ "epoch": 56,
732
+ "loss": 0.2965882046549928,
733
+ "recon_loss": 0.25624091426531476,
734
+ "commit_loss": 0.020173643806985782,
735
+ "embed_loss": 0.020173643806985782,
736
+ "codebook_utilization": 0.247314453125,
737
+ "lr": 8.345653031794289e-05,
738
+ "phase": "full_vq",
739
+ "val_recon_loss": 0.26047537877009463,
740
+ "val_utilization": 0.197021484375,
741
+ "time_sec": 47.16958785057068
742
+ },
743
+ {
744
+ "epoch": 57,
745
+ "loss": 0.2975361960775712,
746
+ "recon_loss": 0.25611695442713944,
747
+ "commit_loss": 0.020709619912154534,
748
+ "embed_loss": 0.020709619912154534,
749
+ "codebook_utilization": 0.244873046875,
750
+ "lr": 8.285358405218651e-05,
751
+ "phase": "full_vq",
752
+ "val_recon_loss": 0.2614493748316398,
753
+ "val_utilization": 0.186279296875,
754
+ "time_sec": 46.03197431564331
755
+ },
756
+ {
757
+ "epoch": 58,
758
+ "loss": 0.29645441677056106,
759
+ "recon_loss": 0.25535813149283915,
760
+ "commit_loss": 0.020548142931040597,
761
+ "embed_loss": 0.020548142931040597,
762
+ "codebook_utilization": 0.24560546875,
763
+ "lr": 8.22421106368085e-05,
764
+ "phase": "full_vq",
765
+ "val_recon_loss": 0.26066550841698277,
766
+ "val_utilization": 0.189208984375,
767
+ "time_sec": 47.509697914123535
768
+ },
769
+ {
770
+ "epoch": 59,
771
+ "loss": 0.2970767763315463,
772
+ "recon_loss": 0.25552325564272266,
773
+ "commit_loss": 0.020776760563546538,
774
+ "embed_loss": 0.020776760563546538,
775
+ "codebook_utilization": 0.2451171875,
776
+ "lr": 8.162226877976883e-05,
777
+ "phase": "full_vq",
778
+ "val_recon_loss": 0.25907388902627504,
779
+ "val_utilization": 0.185302734375,
780
+ "time_sec": 47.40083122253418
781
+ },
782
+ {
783
+ "epoch": 60,
784
+ "loss": 0.29586495663605483,
785
+ "recon_loss": 0.2551090322873172,
786
+ "commit_loss": 0.02037796162653203,
787
+ "embed_loss": 0.02037796162653203,
788
+ "codebook_utilization": 0.2421875,
789
+ "lr": 8.099421936105698e-05,
790
+ "phase": "full_vq",
791
+ "val_recon_loss": 0.25968878200420964,
792
+ "val_utilization": 0.196044921875,
793
+ "time_sec": 47.51141929626465
794
+ },
795
+ {
796
+ "epoch": 61,
797
+ "loss": 0.2953443363601086,
798
+ "recon_loss": 0.2540427975210489,
799
+ "commit_loss": 0.020650770113456483,
800
+ "embed_loss": 0.020650770113456483,
801
+ "codebook_utilization": 0.24169921875,
802
+ "lr": 8.035812539093553e-05,
803
+ "phase": "full_vq",
804
+ "val_recon_loss": 0.2591333996791106,
805
+ "val_utilization": 0.188720703125,
806
+ "time_sec": 48.047051429748535
807
+ },
808
+ {
809
+ "epoch": 62,
810
+ "loss": 0.2951362244054383,
811
+ "recon_loss": 0.25429533977134555,
812
+ "commit_loss": 0.020420441330940117,
813
+ "embed_loss": 0.020420441330940117,
814
+ "codebook_utilization": 0.24072265625,
815
+ "lr": 7.971415196763084e-05,
816
+ "phase": "full_vq",
817
+ "val_recon_loss": 0.2581121852764717,
818
+ "val_utilization": 0.18701171875,
819
+ "time_sec": 47.509090423583984
820
+ },
821
+ {
822
+ "epoch": 63,
823
+ "loss": 0.2947080468430239,
824
+ "recon_loss": 0.2535659708813125,
825
+ "commit_loss": 0.020571036337345253,
826
+ "embed_loss": 0.020571036337345253,
827
+ "codebook_utilization": 0.239990234375,
828
+ "lr": 7.90624662344818e-05,
829
+ "phase": "full_vq",
830
+ "val_recon_loss": 0.25835648064429945,
831
+ "val_utilization": 0.18408203125,
832
+ "time_sec": 45.451547622680664
833
+ },
834
+ {
835
+ "epoch": 64,
836
+ "loss": 0.295073574664546,
837
+ "recon_loss": 0.25326629655033933,
838
+ "commit_loss": 0.020903638363176703,
839
+ "embed_loss": 0.020903638363176703,
840
+ "codebook_utilization": 0.240966796875,
841
+ "lr": 7.840323733655777e-05,
842
+ "phase": "full_vq",
843
+ "val_recon_loss": 0.25807722829855406,
844
+ "val_utilization": 0.19091796875,
845
+ "time_sec": 45.3773295879364
846
+ },
847
+ {
848
+ "epoch": 65,
849
+ "loss": 0.2954476791269639,
850
+ "recon_loss": 0.25291867992457223,
851
+ "commit_loss": 0.02126450186558798,
852
+ "embed_loss": 0.02126450186558798,
853
+ "codebook_utilization": 0.239990234375,
854
+ "lr": 7.773663637675692e-05,
855
+ "phase": "full_vq",
856
+ "val_recon_loss": 0.2571497467847971,
857
+ "val_utilization": 0.189208984375,
858
+ "time_sec": 46.702619552612305
859
+ },
860
+ {
861
+ "epoch": 66,
862
+ "loss": 0.29515332860105176,
863
+ "recon_loss": 0.252620609367595,
864
+ "commit_loss": 0.02126635998195293,
865
+ "embed_loss": 0.02126635998195293,
866
+ "codebook_utilization": 0.2392578125,
867
+ "lr": 7.706283637139655e-05,
868
+ "phase": "full_vq",
869
+ "val_recon_loss": 0.2568894406923881,
870
+ "val_utilization": 0.187744140625,
871
+ "time_sec": 47.769694328308105
872
+ },
873
+ {
874
+ "epoch": 67,
875
+ "loss": 0.295649412800284,
876
+ "recon_loss": 0.25218727191289264,
877
+ "commit_loss": 0.02173107029760585,
878
+ "embed_loss": 0.02173107029760585,
879
+ "codebook_utilization": 0.238525390625,
880
+ "lr": 7.638201220530662e-05,
881
+ "phase": "full_vq",
882
+ "val_recon_loss": 0.25663964794232297,
883
+ "val_utilization": 0.1923828125,
884
+ "time_sec": 48.17647624015808
885
+ },
886
+ {
887
+ "epoch": 68,
888
+ "loss": 0.2957354971006805,
889
+ "recon_loss": 0.25207817817435546,
890
+ "commit_loss": 0.02182865865966853,
891
+ "embed_loss": 0.02182865865966853,
892
+ "codebook_utilization": 0.239501953125,
893
+ "lr": 7.569434058643842e-05,
894
+ "phase": "full_vq",
895
+ "val_recon_loss": 0.2562651519591992,
896
+ "val_utilization": 0.192626953125,
897
+ "time_sec": 45.29482173919678
898
+ },
899
+ {
900
+ "epoch": 69,
901
+ "loss": 0.2953665022756539,
902
+ "recon_loss": 0.2517355958036348,
903
+ "commit_loss": 0.02181545257860539,
904
+ "embed_loss": 0.02181545257860539,
905
+ "codebook_utilization": 0.239990234375,
906
+ "lr": 7.499999999999998e-05,
907
+ "phase": "full_vq",
908
+ "val_recon_loss": 0.2560632870747493,
909
+ "val_utilization": 0.193115234375,
910
+ "time_sec": 48.08019828796387
911
+ },
912
+ {
913
+ "epoch": 70,
914
+ "loss": 0.2958017698690003,
915
+ "recon_loss": 0.25189177136795193,
916
+ "commit_loss": 0.021955000017495716,
917
+ "embed_loss": 0.021955000017495716,
918
+ "codebook_utilization": 0.23828125,
919
+ "lr": 7.429917066213028e-05,
920
+ "phase": "full_vq",
921
+ "val_recon_loss": 0.25639278384355396,
922
+ "val_utilization": 0.19140625,
923
+ "time_sec": 48.016398429870605
924
+ },
925
+ {
926
+ "epoch": 71,
927
+ "loss": 0.2952505992908104,
928
+ "recon_loss": 0.25103825330734253,
929
+ "commit_loss": 0.022106172991733924,
930
+ "embed_loss": 0.022106172991733924,
931
+ "codebook_utilization": 0.2373046875,
932
+ "lr": 7.359203447312409e-05,
933
+ "phase": "full_vq",
934
+ "val_recon_loss": 0.2551685239260013,
935
+ "val_utilization": 0.19091796875,
936
+ "time_sec": 47.73258876800537
937
+ },
938
+ {
939
+ "epoch": 72,
940
+ "loss": 0.29693663646193114,
941
+ "recon_loss": 0.2514823312852897,
942
+ "commit_loss": 0.022727152697888074,
943
+ "embed_loss": 0.022727152697888074,
944
+ "codebook_utilization": 0.236572265625,
945
+ "lr": 7.287877497021975e-05,
946
+ "phase": "full_vq",
947
+ "val_recon_loss": 0.25563192252929395,
948
+ "val_utilization": 0.190185546875,
949
+ "time_sec": 49.903727769851685
950
+ },
951
+ {
952
+ "epoch": 73,
953
+ "loss": 0.29671186500904606,
954
+ "recon_loss": 0.25137501078493457,
955
+ "commit_loss": 0.022668427623370114,
956
+ "embed_loss": 0.022668427623370114,
957
+ "codebook_utilization": 0.23779296875,
958
+ "lr": 7.215957727996207e-05,
959
+ "phase": "full_vq",
960
+ "val_recon_loss": 0.25468500073139483,
961
+ "val_utilization": 0.192138671875,
962
+ "time_sec": 47.38734412193298
963
+ },
964
+ {
965
+ "epoch": 74,
966
+ "loss": 0.297061026096344,
967
+ "recon_loss": 0.2511611797061621,
968
+ "commit_loss": 0.022949924582944196,
969
+ "embed_loss": 0.022949924582944196,
970
+ "codebook_utilization": 0.236328125,
971
+ "lr": 7.143462807015271e-05,
972
+ "phase": "full_vq",
973
+ "val_recon_loss": 0.2554800590643516,
974
+ "val_utilization": 0.184814453125,
975
+ "time_sec": 44.4993736743927
976
+ },
977
+ {
978
+ "epoch": 75,
979
+ "loss": 0.29634231445836084,
980
+ "recon_loss": 0.25069860763409557,
981
+ "commit_loss": 0.022821854690418523,
982
+ "embed_loss": 0.022821854690418523,
983
+ "codebook_utilization": 0.235595703125,
984
+ "lr": 7.07041155014006e-05,
985
+ "phase": "full_vq",
986
+ "val_recon_loss": 0.25494354275556713,
987
+ "val_utilization": 0.190185546875,
988
+ "time_sec": 46.71766662597656
989
+ },
990
+ {
991
+ "epoch": 76,
992
+ "loss": 0.296459870595558,
993
+ "recon_loss": 0.2505088509882198,
994
+ "commit_loss": 0.022975510168893664,
995
+ "embed_loss": 0.022975510168893664,
996
+ "codebook_utilization": 0.2373046875,
997
+ "lr": 6.996822917828478e-05,
998
+ "phase": "full_vq",
999
+ "val_recon_loss": 0.2545968282681245,
1000
+ "val_utilization": 0.192626953125,
1001
+ "time_sec": 47.50995659828186
1002
+ },
1003
+ {
1004
+ "epoch": 77,
1005
+ "loss": 0.29575227522382547,
1006
+ "recon_loss": 0.2501658846934636,
1007
+ "commit_loss": 0.02279319533822583,
1008
+ "embed_loss": 0.02279319533822583,
1009
+ "codebook_utilization": 0.236328125,
1010
+ "lr": 6.922716010014254e-05,
1011
+ "phase": "full_vq",
1012
+ "val_recon_loss": 0.2545246883080556,
1013
+ "val_utilization": 0.18798828125,
1014
+ "time_sec": 48.46063303947449
1015
+ },
1016
+ {
1017
+ "epoch": 78,
1018
+ "loss": 0.2968192123899273,
1019
+ "recon_loss": 0.25013503812107385,
1020
+ "commit_loss": 0.023342087791830887,
1021
+ "embed_loss": 0.023342087791830887,
1022
+ "codebook_utilization": 0.2373046875,
1023
+ "lr": 6.848110061149553e-05,
1024
+ "phase": "full_vq",
1025
+ "val_recon_loss": 0.2544916971371724,
1026
+ "val_utilization": 0.18505859375,
1027
+ "time_sec": 47.71051788330078
1028
+ },
1029
+ {
1030
+ "epoch": 79,
1031
+ "loss": 0.29553129392511707,
1032
+ "recon_loss": 0.2494573739229464,
1033
+ "commit_loss": 0.023036959380203604,
1034
+ "embed_loss": 0.023036959380203604,
1035
+ "codebook_utilization": 0.235595703125,
1036
+ "lr": 6.773024435212678e-05,
1037
+ "phase": "full_vq",
1038
+ "val_recon_loss": 0.25387483491347385,
1039
+ "val_utilization": 0.187255859375,
1040
+ "time_sec": 46.08107137680054
1041
+ },
1042
+ {
1043
+ "epoch": 80,
1044
+ "loss": 0.29670053252986833,
1045
+ "recon_loss": 0.2494831514709136,
1046
+ "commit_loss": 0.023608688849444483,
1047
+ "embed_loss": 0.023608688849444483,
1048
+ "codebook_utilization": 0.233154296875,
1049
+ "lr": 6.697478620682138e-05,
1050
+ "phase": "full_vq",
1051
+ "val_recon_loss": 0.2537316714341824,
1052
+ "val_utilization": 0.1875,
1053
+ "time_sec": 45.84971261024475
1054
+ },
1055
+ {
1056
+ "epoch": 81,
1057
+ "loss": 0.2959857951192295,
1058
+ "recon_loss": 0.2489085025062748,
1059
+ "commit_loss": 0.023538647109971327,
1060
+ "embed_loss": 0.023538647109971327,
1061
+ "codebook_utilization": 0.232177734375,
1062
+ "lr": 6.621492225478415e-05,
1063
+ "phase": "full_vq",
1064
+ "val_recon_loss": 0.2539532551398644,
1065
+ "val_utilization": 0.1884765625,
1066
+ "time_sec": 46.758139848709106
1067
+ },
1068
+ {
1069
+ "epoch": 82,
1070
+ "loss": 0.2976532619373471,
1071
+ "recon_loss": 0.24936814956805287,
1072
+ "commit_loss": 0.02414255665943903,
1073
+ "embed_loss": 0.02414255665943903,
1074
+ "codebook_utilization": 0.232421875,
1075
+ "lr": 6.545084971874738e-05,
1076
+ "phase": "full_vq",
1077
+ "val_recon_loss": 0.25328402794324434,
1078
+ "val_utilization": 0.18603515625,
1079
+ "time_sec": 47.337982177734375
1080
+ },
1081
+ {
1082
+ "epoch": 83,
1083
+ "loss": 0.2973826387349297,
1084
+ "recon_loss": 0.24920354228393704,
1085
+ "commit_loss": 0.024089547531569704,
1086
+ "embed_loss": 0.024089547531569704,
1087
+ "codebook_utilization": 0.233642578125,
1088
+ "lr": 6.468276691378155e-05,
1089
+ "phase": "full_vq",
1090
+ "val_recon_loss": 0.2519659594847606,
1091
+ "val_utilization": 0.185791015625,
1092
+ "time_sec": 48.023863315582275
1093
+ },
1094
+ {
1095
+ "epoch": 84,
1096
+ "loss": 0.29639680712830785,
1097
+ "recon_loss": 0.24831042745534113,
1098
+ "commit_loss": 0.024043192100875518,
1099
+ "embed_loss": 0.024043192100875518,
1100
+ "codebook_utilization": 0.232666015625,
1101
+ "lr": 6.391087319582264e-05,
1102
+ "phase": "full_vq",
1103
+ "val_recon_loss": 0.2526073604822159,
1104
+ "val_utilization": 0.185302734375,
1105
+ "time_sec": 46.68369913101196
1106
+ },
1107
+ {
1108
+ "epoch": 85,
1109
+ "loss": 0.29832304400556225,
1110
+ "recon_loss": 0.24883184041462691,
1111
+ "commit_loss": 0.024745601612855408,
1112
+ "embed_loss": 0.024745601612855408,
1113
+ "codebook_utilization": 0.230712890625,
1114
+ "lr": 6.313536890992936e-05,
1115
+ "phase": "full_vq",
1116
+ "val_recon_loss": 0.2527536623753034,
1117
+ "val_utilization": 0.186767578125,
1118
+ "time_sec": 45.81481218338013
1119
+ },
1120
+ {
1121
+ "epoch": 86,
1122
+ "loss": 0.2982981380294351,
1123
+ "recon_loss": 0.24891039641464457,
1124
+ "commit_loss": 0.024693871099574893,
1125
+ "embed_loss": 0.024693871099574893,
1126
+ "codebook_utilization": 0.233642578125,
1127
+ "lr": 6.23564553382835e-05,
1128
+ "phase": "full_vq",
1129
+ "val_recon_loss": 0.25296917443092054,
1130
+ "val_utilization": 0.19091796875,
1131
+ "time_sec": 46.41946482658386
1132
+ },
1133
+ {
1134
+ "epoch": 87,
1135
+ "loss": 0.29725263048620787,
1136
+ "recon_loss": 0.2481701578579697,
1137
+ "commit_loss": 0.024541236861955886,
1138
+ "embed_loss": 0.024541236861955886,
1139
+ "codebook_utilization": 0.232177734375,
1140
+ "lr": 6.157433464794716e-05,
1141
+ "phase": "full_vq",
1142
+ "val_recon_loss": 0.25219068160423863,
1143
+ "val_utilization": 0.184814453125,
1144
+ "time_sec": 47.36266231536865
1145
+ },
1146
+ {
1147
+ "epoch": 88,
1148
+ "loss": 0.29676519129790513,
1149
+ "recon_loss": 0.24771483359383603,
1150
+ "commit_loss": 0.024525179472916266,
1151
+ "embed_loss": 0.024525179472916266,
1152
+ "codebook_utilization": 0.23291015625,
1153
+ "lr": 6.0789209838390326e-05,
1154
+ "phase": "full_vq",
1155
+ "val_recon_loss": 0.25175508054403156,
1156
+ "val_utilization": 0.1826171875,
1157
+ "time_sec": 47.208651542663574
1158
+ },
1159
+ {
1160
+ "epoch": 89,
1161
+ "loss": 0.2978205289326462,
1162
+ "recon_loss": 0.24747842316534005,
1163
+ "commit_loss": 0.02517105018099149,
1164
+ "embed_loss": 0.02517105018099149,
1165
+ "codebook_utilization": 0.232421875,
1166
+ "lr": 6.000128468880224e-05,
1167
+ "phase": "full_vq",
1168
+ "val_recon_loss": 0.25257671910982865,
1169
+ "val_utilization": 0.184814453125,
1170
+ "time_sec": 46.57335686683655
1171
+ },
1172
+ {
1173
+ "epoch": 90,
1174
+ "loss": 0.29792727909836114,
1175
+ "recon_loss": 0.2476864559393303,
1176
+ "commit_loss": 0.025120412090829776,
1177
+ "embed_loss": 0.025120412090829776,
1178
+ "codebook_utilization": 0.229248046875,
1179
+ "lr": 5.9210763705200585e-05,
1180
+ "phase": "full_vq",
1181
+ "val_recon_loss": 0.2515146594781142,
1182
+ "val_utilization": 0.185546875,
1183
+ "time_sec": 46.14790081977844
1184
+ },
1185
+ {
1186
+ "epoch": 91,
1187
+ "loss": 0.2983899794372858,
1188
+ "recon_loss": 0.24762496498285555,
1189
+ "commit_loss": 0.025382505474137326,
1190
+ "embed_loss": 0.025382505474137326,
1191
+ "codebook_utilization": 0.2314453125,
1192
+ "lr": 5.841785206735194e-05,
1193
+ "phase": "full_vq",
1194
+ "val_recon_loss": 0.2524160181100552,
1195
+ "val_utilization": 0.180419921875,
1196
+ "time_sec": 47.52765107154846
1197
+ },
1198
+ {
1199
+ "epoch": 92,
1200
+ "loss": 0.29805802889898714,
1201
+ "recon_loss": 0.24733802442457162,
1202
+ "commit_loss": 0.02536000194502812,
1203
+ "embed_loss": 0.02536000194502812,
1204
+ "codebook_utilization": 0.22998046875,
1205
+ "lr": 5.7622755575517285e-05,
1206
+ "phase": "full_vq",
1207
+ "val_recon_loss": 0.2506679456967574,
1208
+ "val_utilization": 0.183837890625,
1209
+ "time_sec": 47.61260223388672
1210
+ },
1211
+ {
1212
+ "epoch": 93,
1213
+ "loss": 0.298675339011585,
1214
+ "recon_loss": 0.2470666514892204,
1215
+ "commit_loss": 0.02580434339595776,
1216
+ "embed_loss": 0.02580434339595776,
1217
+ "codebook_utilization": 0.230224609375,
1218
+ "lr": 5.6825680597036586e-05,
1219
+ "phase": "full_vq",
1220
+ "val_recon_loss": 0.2510277766447801,
1221
+ "val_utilization": 0.1904296875,
1222
+ "time_sec": 45.46219205856323
1223
+ },
1224
+ {
1225
+ "epoch": 94,
1226
+ "loss": 0.2984109982555988,
1227
+ "recon_loss": 0.2473060015369864,
1228
+ "commit_loss": 0.025552497117542754,
1229
+ "embed_loss": 0.025552497117542754,
1230
+ "codebook_utilization": 0.229736328125,
1231
+ "lr": 5.602683401276615e-05,
1232
+ "phase": "full_vq",
1233
+ "val_recon_loss": 0.2512832169349377,
1234
+ "val_utilization": 0.1865234375,
1235
+ "time_sec": 46.23483443260193
1236
+ },
1237
+ {
1238
+ "epoch": 95,
1239
+ "loss": 0.2980131354986453,
1240
+ "recon_loss": 0.24664961035345115,
1241
+ "commit_loss": 0.025681763230001226,
1242
+ "embed_loss": 0.025681763230001226,
1243
+ "codebook_utilization": 0.229736328125,
1244
+ "lr": 5.522642316338268e-05,
1245
+ "phase": "full_vq",
1246
+ "val_recon_loss": 0.2509289475587698,
1247
+ "val_utilization": 0.18798828125,
1248
+ "time_sec": 47.12396693229675
1249
+ },
1250
+ {
1251
+ "epoch": 96,
1252
+ "loss": 0.30049163745898827,
1253
+ "recon_loss": 0.2471260504395354,
1254
+ "commit_loss": 0.026682794970624587,
1255
+ "embed_loss": 0.026682794970624587,
1256
+ "codebook_utilization": 0.229248046875,
1257
+ "lr": 5.4424655795567935e-05,
1258
+ "phase": "full_vq",
1259
+ "val_recon_loss": 0.2515842364384578,
1260
+ "val_utilization": 0.184326171875,
1261
+ "time_sec": 47.4691379070282
1262
+ },
1263
+ {
1264
+ "epoch": 97,
1265
+ "loss": 0.2983791185360329,
1266
+ "recon_loss": 0.2468152753278321,
1267
+ "commit_loss": 0.02578192259020665,
1268
+ "embed_loss": 0.02578192259020665,
1269
+ "codebook_utilization": 0.23046875,
1270
+ "lr": 5.3621740008088126e-05,
1271
+ "phase": "full_vq",
1272
+ "val_recon_loss": 0.2509624602702948,
1273
+ "val_utilization": 0.18603515625,
1274
+ "time_sec": 48.37575817108154
1275
+ },
1276
+ {
1277
+ "epoch": 98,
1278
+ "loss": 0.2991181933412365,
1279
+ "recon_loss": 0.24653221433069192,
1280
+ "commit_loss": 0.026292991660097065,
1281
+ "embed_loss": 0.026292991660097065,
1282
+ "codebook_utilization": 0.2294921875,
1283
+ "lr": 5.281788419778186e-05,
1284
+ "phase": "full_vq",
1285
+ "val_recon_loss": 0.25135975961501783,
1286
+ "val_utilization": 0.188232421875,
1287
+ "time_sec": 49.54656410217285
1288
+ },
1289
+ {
1290
+ "epoch": 99,
1291
+ "loss": 0.2981439980806089,
1292
+ "recon_loss": 0.24589644518553042,
1293
+ "commit_loss": 0.026123775607522798,
1294
+ "embed_loss": 0.026123775607522798,
1295
+ "codebook_utilization": 0.229736328125,
1296
+ "lr": 5.201329700547076e-05,
1297
+ "phase": "full_vq",
1298
+ "val_recon_loss": 0.25068271618623,
1299
+ "val_utilization": 0.186279296875,
1300
+ "time_sec": 49.6520459651947
1301
+ },
1302
+ {
1303
+ "epoch": 100,
1304
+ "loss": 0.2999950992126091,
1305
+ "recon_loss": 0.24611248368141697,
1306
+ "commit_loss": 0.02694130736384906,
1307
+ "embed_loss": 0.02694130736384906,
1308
+ "codebook_utilization": 0.228759765625,
1309
+ "lr": 5.120818726180663e-05,
1310
+ "phase": "full_vq",
1311
+ "val_recon_loss": 0.2501685596429385,
1312
+ "val_utilization": 0.182861328125,
1313
+ "time_sec": 49.29077219963074
1314
+ },
1315
+ {
1316
+ "epoch": 101,
1317
+ "loss": 0.30049710764604454,
1318
+ "recon_loss": 0.24640378239108066,
1319
+ "commit_loss": 0.027046663175318755,
1320
+ "embed_loss": 0.027046663175318755,
1321
+ "codebook_utilization": 0.228515625,
1322
+ "lr": 5.040276393306951e-05,
1323
+ "phase": "full_vq",
1324
+ "val_recon_loss": 0.2505534818539253,
1325
+ "val_utilization": 0.189453125,
1326
+ "time_sec": 48.92568397521973
1327
+ },
1328
+ {
1329
+ "epoch": 102,
1330
+ "loss": 0.29963186207939596,
1331
+ "recon_loss": 0.24578097082820594,
1332
+ "commit_loss": 0.026925445771684833,
1333
+ "embed_loss": 0.026925445771684833,
1334
+ "codebook_utilization": 0.228515625,
1335
+ "lr": 4.9597236066930516e-05,
1336
+ "phase": "full_vq",
1337
+ "val_recon_loss": 0.24981138454033777,
1338
+ "val_utilization": 0.18798828125,
1339
+ "time_sec": 49.33101749420166
1340
+ },
1341
+ {
1342
+ "epoch": 103,
1343
+ "loss": 0.29956874777289,
1344
+ "recon_loss": 0.24556882241192987,
1345
+ "commit_loss": 0.026999962753524967,
1346
+ "embed_loss": 0.026999962753524967,
1347
+ "codebook_utilization": 0.22802734375,
1348
+ "lr": 4.879181273819339e-05,
1349
+ "phase": "full_vq",
1350
+ "val_recon_loss": 0.25028561743406147,
1351
+ "val_utilization": 0.185546875,
1352
+ "time_sec": 48.70212149620056
1353
+ },
1354
+ {
1355
+ "epoch": 104,
1356
+ "loss": 0.30101509012427985,
1357
+ "recon_loss": 0.24612126309497684,
1358
+ "commit_loss": 0.027446913697263774,
1359
+ "embed_loss": 0.027446913697263774,
1360
+ "codebook_utilization": 0.22802734375,
1361
+ "lr": 4.798670299452925e-05,
1362
+ "phase": "full_vq",
1363
+ "val_recon_loss": 0.25050369936686295,
1364
+ "val_utilization": 0.178955078125,
1365
+ "time_sec": 48.56446051597595
1366
+ },
1367
+ {
1368
+ "epoch": 105,
1369
+ "loss": 0.299971297675488,
1370
+ "recon_loss": 0.24605776340353722,
1371
+ "commit_loss": 0.026956765492464983,
1372
+ "embed_loss": 0.026956765492464983,
1373
+ "codebook_utilization": 0.22900390625,
1374
+ "lr": 4.718211580221814e-05,
1375
+ "phase": "full_vq",
1376
+ "val_recon_loss": 0.2500560008562528,
1377
+ "val_utilization": 0.183349609375,
1378
+ "time_sec": 49.97710371017456
1379
+ },
1380
+ {
1381
+ "epoch": 106,
1382
+ "loss": 0.2991456208275814,
1383
+ "recon_loss": 0.24511409653168098,
1384
+ "commit_loss": 0.02701576196533792,
1385
+ "embed_loss": 0.02701576196533792,
1386
+ "codebook_utilization": 0.2275390625,
1387
+ "lr": 4.637825999191189e-05,
1388
+ "phase": "full_vq",
1389
+ "val_recon_loss": 0.2502270879653784,
1390
+ "val_utilization": 0.18701171875,
1391
+ "time_sec": 49.39626979827881
1392
+ },
1393
+ {
1394
+ "epoch": 107,
1395
+ "loss": 0.2991793132295795,
1396
+ "recon_loss": 0.2449595960331898,
1397
+ "commit_loss": 0.027109859036464318,
1398
+ "embed_loss": 0.027109859036464318,
1399
+ "codebook_utilization": 0.227294921875,
1400
+ "lr": 4.557534420443209e-05,
1401
+ "phase": "full_vq",
1402
+ "val_recon_loss": 0.24889626296666953,
1403
+ "val_utilization": 0.185791015625,
1404
+ "time_sec": 48.22566032409668
1405
+ },
1406
+ {
1407
+ "epoch": 108,
1408
+ "loss": 0.30064308000545875,
1409
+ "recon_loss": 0.24495375390146293,
1410
+ "commit_loss": 0.027844665133777785,
1411
+ "embed_loss": 0.027844665133777785,
1412
+ "codebook_utilization": 0.22802734375,
1413
+ "lr": 4.477357683661733e-05,
1414
+ "phase": "full_vq",
1415
+ "val_recon_loss": 0.24943268528351417,
1416
+ "val_utilization": 0.186279296875,
1417
+ "time_sec": 49.18807530403137
1418
+ },
1419
+ {
1420
+ "epoch": 109,
1421
+ "loss": 0.30050324984625276,
1422
+ "recon_loss": 0.24478166652660743,
1423
+ "commit_loss": 0.027860790052834677,
1424
+ "embed_loss": 0.027860790052834677,
1425
+ "codebook_utilization": 0.228759765625,
1426
+ "lr": 4.397316598723385e-05,
1427
+ "phase": "full_vq",
1428
+ "val_recon_loss": 0.2500851463813048,
1429
+ "val_utilization": 0.18212890625,
1430
+ "time_sec": 49.43425464630127
1431
+ },
1432
+ {
1433
+ "epoch": 110,
1434
+ "loss": 0.2998181587340785,
1435
+ "recon_loss": 0.24475371633090226,
1436
+ "commit_loss": 0.027532222845098552,
1437
+ "embed_loss": 0.027532222845098552,
1438
+ "codebook_utilization": 0.225830078125,
1439
+ "lr": 4.317431940296343e-05,
1440
+ "phase": "full_vq",
1441
+ "val_recon_loss": 0.24926420129262483,
1442
+ "val_utilization": 0.188720703125,
1443
+ "time_sec": 48.50480031967163
1444
+ },
1445
+ {
1446
+ "epoch": 111,
1447
+ "loss": 0.30019010574209926,
1448
+ "recon_loss": 0.24445770271852904,
1449
+ "commit_loss": 0.02786620184048718,
1450
+ "embed_loss": 0.02786620184048718,
1451
+ "codebook_utilization": 0.2265625,
1452
+ "lr": 4.237724442448274e-05,
1453
+ "phase": "full_vq",
1454
+ "val_recon_loss": 0.24935368505808023,
1455
+ "val_utilization": 0.187255859375,
1456
+ "time_sec": 47.10006523132324
1457
+ },
1458
+ {
1459
+ "epoch": 112,
1460
+ "loss": 0.3010977863096723,
1461
+ "recon_loss": 0.24472794900922215,
1462
+ "commit_loss": 0.028184919745898714,
1463
+ "embed_loss": 0.028184919745898714,
1464
+ "codebook_utilization": 0.225341796875,
1465
+ "lr": 4.158214793264808e-05,
1466
+ "phase": "full_vq",
1467
+ "val_recon_loss": 0.249008270410391,
1468
+ "val_utilization": 0.185302734375,
1469
+ "time_sec": 48.75494933128357
1470
+ },
1471
+ {
1472
+ "epoch": 113,
1473
+ "loss": 0.30125294771848943,
1474
+ "recon_loss": 0.24450724761860043,
1475
+ "commit_loss": 0.0283728501960343,
1476
+ "embed_loss": 0.0283728501960343,
1477
+ "codebook_utilization": 0.2275390625,
1478
+ "lr": 4.078923629479945e-05,
1479
+ "phase": "full_vq",
1480
+ "val_recon_loss": 0.24862084824305314,
1481
+ "val_utilization": 0.187255859375,
1482
+ "time_sec": 49.49637579917908
1483
+ },
1484
+ {
1485
+ "epoch": 114,
1486
+ "loss": 0.30052362820681405,
1487
+ "recon_loss": 0.24406547055524938,
1488
+ "commit_loss": 0.02822907999450085,
1489
+ "embed_loss": 0.02822907999450085,
1490
+ "codebook_utilization": 0.225341796875,
1491
+ "lr": 3.9998715311197785e-05,
1492
+ "phase": "full_vq",
1493
+ "val_recon_loss": 0.24960095951190361,
1494
+ "val_utilization": 0.18359375,
1495
+ "time_sec": 48.85350823402405
1496
+ },
1497
+ {
1498
+ "epoch": 115,
1499
+ "loss": 0.30091238314030216,
1500
+ "recon_loss": 0.2442388341707342,
1501
+ "commit_loss": 0.028336774411739086,
1502
+ "embed_loss": 0.028336774411739086,
1503
+ "codebook_utilization": 0.224609375,
1504
+ "lr": 3.9210790161609705e-05,
1505
+ "phase": "full_vq",
1506
+ "val_recon_loss": 0.248817842740279,
1507
+ "val_utilization": 0.1845703125,
1508
+ "time_sec": 50.234500885009766
1509
+ },
1510
+ {
1511
+ "epoch": 116,
1512
+ "loss": 0.30179913721832574,
1513
+ "recon_loss": 0.2441415824726516,
1514
+ "commit_loss": 0.028828777920673874,
1515
+ "embed_loss": 0.028828777920673874,
1516
+ "codebook_utilization": 0.22607421875,
1517
+ "lr": 3.8425665352052864e-05,
1518
+ "phase": "full_vq",
1519
+ "val_recon_loss": 0.24841140898374411,
1520
+ "val_utilization": 0.1826171875,
1521
+ "time_sec": 50.250232219696045
1522
+ },
1523
+ {
1524
+ "epoch": 117,
1525
+ "loss": 0.30132410865204007,
1526
+ "recon_loss": 0.24388884563072055,
1527
+ "commit_loss": 0.02871763278894565,
1528
+ "embed_loss": 0.02871763278894565,
1529
+ "codebook_utilization": 0.225341796875,
1530
+ "lr": 3.764354466171653e-05,
1531
+ "phase": "full_vq",
1532
+ "val_recon_loss": 0.24872419925836417,
1533
+ "val_utilization": 0.186767578125,
1534
+ "time_sec": 48.885910511016846
1535
+ },
1536
+ {
1537
+ "epoch": 118,
1538
+ "loss": 0.30116594305225447,
1539
+ "recon_loss": 0.2436809016793382,
1540
+ "commit_loss": 0.028742521380384762,
1541
+ "embed_loss": 0.028742521380384762,
1542
+ "codebook_utilization": 0.225341796875,
1543
+ "lr": 3.686463109007067e-05,
1544
+ "phase": "full_vq",
1545
+ "val_recon_loss": 0.24823532196191642,
1546
+ "val_utilization": 0.1865234375,
1547
+ "time_sec": 47.51360464096069
1548
+ },
1549
+ {
1550
+ "epoch": 119,
1551
+ "loss": 0.30173085542286143,
1552
+ "recon_loss": 0.2437804443579094,
1553
+ "commit_loss": 0.028975205094206567,
1554
+ "embed_loss": 0.028975205094206567,
1555
+ "codebook_utilization": 0.224609375,
1556
+ "lr": 3.6089126804177374e-05,
1557
+ "phase": "full_vq",
1558
+ "val_recon_loss": 0.2487882673740387,
1559
+ "val_utilization": 0.18359375,
1560
+ "time_sec": 48.14443373680115
1561
+ },
1562
+ {
1563
+ "epoch": 120,
1564
+ "loss": 0.3025599145421795,
1565
+ "recon_loss": 0.2440522172871758,
1566
+ "commit_loss": 0.029253848262277303,
1567
+ "embed_loss": 0.029253848262277303,
1568
+ "codebook_utilization": 0.223388671875,
1569
+ "lr": 3.531723308621847e-05,
1570
+ "phase": "full_vq",
1571
+ "val_recon_loss": 0.24821116030216217,
1572
+ "val_utilization": 0.18359375,
1573
+ "time_sec": 48.07896876335144
1574
+ },
1575
+ {
1576
+ "epoch": 121,
1577
+ "loss": 0.3019287703084011,
1578
+ "recon_loss": 0.24345431374568566,
1579
+ "commit_loss": 0.029237228719627157,
1580
+ "embed_loss": 0.029237228719627157,
1581
+ "codebook_utilization": 0.22314453125,
1582
+ "lr": 3.4549150281252636e-05,
1583
+ "phase": "full_vq",
1584
+ "val_recon_loss": 0.24909190718944257,
1585
+ "val_utilization": 0.185302734375,
1586
+ "time_sec": 47.84167146682739
1587
+ },
1588
+ {
1589
+ "epoch": 122,
1590
+ "loss": 0.3015232845848682,
1591
+ "recon_loss": 0.2433141160245035,
1592
+ "commit_loss": 0.02910458274623927,
1593
+ "embed_loss": 0.02910458274623927,
1594
+ "codebook_utilization": 0.22314453125,
1595
+ "lr": 3.3785077745215873e-05,
1596
+ "phase": "full_vq",
1597
+ "val_recon_loss": 0.24815640311974746,
1598
+ "val_utilization": 0.183837890625,
1599
+ "time_sec": 48.776561975479126
1600
+ },
1601
+ {
1602
+ "epoch": 123,
1603
+ "loss": 0.30234258315142465,
1604
+ "recon_loss": 0.243118543250888,
1605
+ "commit_loss": 0.02961201991374586,
1606
+ "embed_loss": 0.02961201991374586,
1607
+ "codebook_utilization": 0.221923828125,
1608
+ "lr": 3.302521379317865e-05,
1609
+ "phase": "full_vq",
1610
+ "val_recon_loss": 0.24863206423245943,
1611
+ "val_utilization": 0.18310546875,
1612
+ "time_sec": 48.30086588859558
1613
+ },
1614
+ {
1615
+ "epoch": 124,
1616
+ "loss": 0.30237510040694593,
1617
+ "recon_loss": 0.2432328620377709,
1618
+ "commit_loss": 0.0295711195132896,
1619
+ "embed_loss": 0.0295711195132896,
1620
+ "codebook_utilization": 0.221923828125,
1621
+ "lr": 3.226975564787323e-05,
1622
+ "phase": "full_vq",
1623
+ "val_recon_loss": 0.24849343872987306,
1624
+ "val_utilization": 0.187744140625,
1625
+ "time_sec": 48.593151330947876
1626
+ },
1627
+ {
1628
+ "epoch": 125,
1629
+ "loss": 0.30339572184226093,
1630
+ "recon_loss": 0.24364687648474,
1631
+ "commit_loss": 0.029874423043984993,
1632
+ "embed_loss": 0.029874423043984993,
1633
+ "codebook_utilization": 0.221923828125,
1634
+ "lr": 3.151889938850445e-05,
1635
+ "phase": "full_vq",
1636
+ "val_recon_loss": 0.2485118691737835,
1637
+ "val_utilization": 0.182373046875,
1638
+ "time_sec": 47.91917824745178
1639
+ },
1640
+ {
1641
+ "epoch": 126,
1642
+ "loss": 0.3026172232394125,
1643
+ "recon_loss": 0.2432735387016745,
1644
+ "commit_loss": 0.029671843145407883,
1645
+ "embed_loss": 0.029671843145407883,
1646
+ "codebook_utilization": 0.2236328125,
1647
+ "lr": 3.0772839899857464e-05,
1648
+ "phase": "full_vq",
1649
+ "val_recon_loss": 0.24815468719372383,
1650
+ "val_utilization": 0.1796875,
1651
+ "time_sec": 48.62407684326172
1652
+ },
1653
+ {
1654
+ "epoch": 127,
1655
+ "loss": 0.30274081697650984,
1656
+ "recon_loss": 0.24313928712816799,
1657
+ "commit_loss": 0.02980076638506908,
1658
+ "embed_loss": 0.02980076638506908,
1659
+ "codebook_utilization": 0.22265625,
1660
+ "lr": 3.003177082171522e-05,
1661
+ "phase": "full_vq",
1662
+ "val_recon_loss": 0.2485698564694478,
1663
+ "val_utilization": 0.182861328125,
1664
+ "time_sec": 49.14924669265747
1665
+ },
1666
+ {
1667
+ "epoch": 128,
1668
+ "loss": 0.3033904345596538,
1669
+ "recon_loss": 0.24353516803068273,
1670
+ "commit_loss": 0.029927634944518406,
1671
+ "embed_loss": 0.029927634944518406,
1672
+ "codebook_utilization": 0.222900390625,
1673
+ "lr": 2.92958844985994e-05,
1674
+ "phase": "full_vq",
1675
+ "val_recon_loss": 0.24788776957071745,
1676
+ "val_utilization": 0.186767578125,
1677
+ "time_sec": 48.097257137298584
1678
+ },
1679
+ {
1680
+ "epoch": 129,
1681
+ "loss": 0.30307839956938054,
1682
+ "recon_loss": 0.24338933416441375,
1683
+ "commit_loss": 0.02984453153376486,
1684
+ "embed_loss": 0.02984453153376486,
1685
+ "codebook_utilization": 0.221923828125,
1686
+ "lr": 2.8565371929847304e-05,
1687
+ "phase": "full_vq",
1688
+ "val_recon_loss": 0.2479520451564055,
1689
+ "val_utilization": 0.1826171875,
1690
+ "time_sec": 49.522122859954834
1691
+ },
1692
+ {
1693
+ "epoch": 130,
1694
+ "loss": 0.30236775384229775,
1695
+ "recon_loss": 0.2426743489854476,
1696
+ "commit_loss": 0.029846701917110707,
1697
+ "embed_loss": 0.029846701917110707,
1698
+ "codebook_utilization": 0.221923828125,
1699
+ "lr": 2.7840422720037937e-05,
1700
+ "phase": "full_vq",
1701
+ "val_recon_loss": 0.24771552017101875,
1702
+ "val_utilization": 0.18359375,
1703
+ "time_sec": 47.526249170303345
1704
+ },
1705
+ {
1706
+ "epoch": 131,
1707
+ "loss": 0.30273402496880175,
1708
+ "recon_loss": 0.24275574701673844,
1709
+ "commit_loss": 0.02998913974300319,
1710
+ "embed_loss": 0.02998913974300319,
1711
+ "codebook_utilization": 0.221923828125,
1712
+ "lr": 2.7121225029780234e-05,
1713
+ "phase": "full_vq",
1714
+ "val_recon_loss": 0.24802619677323562,
1715
+ "val_utilization": 0.182861328125,
1716
+ "time_sec": 48.135313987731934
1717
+ },
1718
+ {
1719
+ "epoch": 132,
1720
+ "loss": 0.30300195252194123,
1721
+ "recon_loss": 0.24287309716729558,
1722
+ "commit_loss": 0.030064427202530934,
1723
+ "embed_loss": 0.030064427202530934,
1724
+ "codebook_utilization": 0.221923828125,
1725
+ "lr": 2.6407965526875892e-05,
1726
+ "phase": "full_vq",
1727
+ "val_recon_loss": 0.24807491898536682,
1728
+ "val_utilization": 0.184326171875,
1729
+ "time_sec": 49.173460721969604
1730
+ },
1731
+ {
1732
+ "epoch": 133,
1733
+ "loss": 0.3028263771066479,
1734
+ "recon_loss": 0.24281710386276245,
1735
+ "commit_loss": 0.0300046365488978,
1736
+ "embed_loss": 0.0300046365488978,
1737
+ "codebook_utilization": 0.2216796875,
1738
+ "lr": 2.570082933786969e-05,
1739
+ "phase": "full_vq",
1740
+ "val_recon_loss": 0.24734561145305634,
1741
+ "val_utilization": 0.18310546875,
1742
+ "time_sec": 48.65791296958923
1743
+ },
1744
+ {
1745
+ "epoch": 134,
1746
+ "loss": 0.3030599104423149,
1747
+ "recon_loss": 0.24260777030505387,
1748
+ "commit_loss": 0.030226069411226346,
1749
+ "embed_loss": 0.030226069411226346,
1750
+ "codebook_utilization": 0.221923828125,
1751
+ "lr": 2.5000000000000008e-05,
1752
+ "phase": "full_vq",
1753
+ "val_recon_loss": 0.24709115005456483,
1754
+ "val_utilization": 0.18359375,
1755
+ "time_sec": 48.861212730407715
1756
+ },
1757
+ {
1758
+ "epoch": 135,
1759
+ "loss": 0.30314985443564024,
1760
+ "recon_loss": 0.24275903754374561,
1761
+ "commit_loss": 0.03019540844594731,
1762
+ "embed_loss": 0.03019540844594731,
1763
+ "codebook_utilization": 0.2216796875,
1764
+ "lr": 2.4305659413561568e-05,
1765
+ "phase": "full_vq",
1766
+ "val_recon_loss": 0.24765535157460433,
1767
+ "val_utilization": 0.18505859375,
1768
+ "time_sec": 48.81813931465149
1769
+ },
1770
+ {
1771
+ "epoch": 136,
1772
+ "loss": 0.304226009869108,
1773
+ "recon_loss": 0.2426794264830795,
1774
+ "commit_loss": 0.03077329216780616,
1775
+ "embed_loss": 0.03077329216780616,
1776
+ "codebook_utilization": 0.220947265625,
1777
+ "lr": 2.3617987794693356e-05,
1778
+ "phase": "full_vq",
1779
+ "val_recon_loss": 0.2475110269509829,
1780
+ "val_utilization": 0.18408203125,
1781
+ "time_sec": 49.449166774749756
1782
+ },
1783
+ {
1784
+ "epoch": 137,
1785
+ "loss": 0.3034880996919146,
1786
+ "recon_loss": 0.24238746511001213,
1787
+ "commit_loss": 0.030550315830053066,
1788
+ "embed_loss": 0.030550315830053066,
1789
+ "codebook_utilization": 0.220947265625,
1790
+ "lr": 2.2937163628603435e-05,
1791
+ "phase": "full_vq",
1792
+ "val_recon_loss": 0.2475822063592764,
1793
+ "val_utilization": 0.182861328125,
1794
+ "time_sec": 50.61092662811279
1795
+ },
1796
+ {
1797
+ "epoch": 138,
1798
+ "loss": 0.30376631021499634,
1799
+ "recon_loss": 0.24225654496866114,
1800
+ "commit_loss": 0.0307548830979595,
1801
+ "embed_loss": 0.0307548830979595,
1802
+ "codebook_utilization": 0.221435546875,
1803
+ "lr": 2.2263363623243054e-05,
1804
+ "phase": "full_vq",
1805
+ "val_recon_loss": 0.24704946119051713,
1806
+ "val_utilization": 0.18359375,
1807
+ "time_sec": 50.30013942718506
1808
+ },
1809
+ {
1810
+ "epoch": 139,
1811
+ "loss": 0.30381928355086085,
1812
+ "recon_loss": 0.24237151853009767,
1813
+ "commit_loss": 0.030723882473859133,
1814
+ "embed_loss": 0.030723882473859133,
1815
+ "codebook_utilization": 0.2216796875,
1816
+ "lr": 2.1596762663442214e-05,
1817
+ "phase": "full_vq",
1818
+ "val_recon_loss": 0.24778547539160803,
1819
+ "val_utilization": 0.18359375,
1820
+ "time_sec": 49.365148067474365
1821
+ },
1822
+ {
1823
+ "epoch": 140,
1824
+ "loss": 0.30365436451107847,
1825
+ "recon_loss": 0.24222724256562253,
1826
+ "commit_loss": 0.030713559584874733,
1827
+ "embed_loss": 0.030713559584874733,
1828
+ "codebook_utilization": 0.220703125,
1829
+ "lr": 2.0937533765518167e-05,
1830
+ "phase": "full_vq",
1831
+ "val_recon_loss": 0.2473998528260451,
1832
+ "val_utilization": 0.181640625,
1833
+ "time_sec": 46.48232817649841
1834
+ },
1835
+ {
1836
+ "epoch": 141,
1837
+ "loss": 0.30363455531643885,
1838
+ "recon_loss": 0.2422852092513851,
1839
+ "commit_loss": 0.030674672630779883,
1840
+ "embed_loss": 0.030674672630779883,
1841
+ "codebook_utilization": 0.220947265625,
1842
+ "lr": 2.0285848032369133e-05,
1843
+ "phase": "full_vq",
1844
+ "val_recon_loss": 0.24757225926105791,
1845
+ "val_utilization": 0.183837890625,
1846
+ "time_sec": 45.080886125564575
1847
+ },
1848
+ {
1849
+ "epoch": 142,
1850
+ "loss": 0.30396323110543044,
1851
+ "recon_loss": 0.24214003395800496,
1852
+ "commit_loss": 0.030911598865892374,
1853
+ "embed_loss": 0.030911598865892374,
1854
+ "codebook_utilization": 0.220703125,
1855
+ "lr": 1.964187460906444e-05,
1856
+ "phase": "full_vq",
1857
+ "val_recon_loss": 0.24763313623575064,
1858
+ "val_utilization": 0.181884765625,
1859
+ "time_sec": 46.7608060836792
1860
+ },
1861
+ {
1862
+ "epoch": 143,
1863
+ "loss": 0.3039931374437669,
1864
+ "recon_loss": 0.24211384940381145,
1865
+ "commit_loss": 0.030939644567814527,
1866
+ "embed_loss": 0.030939644567814527,
1867
+ "codebook_utilization": 0.2216796875,
1868
+ "lr": 1.9005780638942995e-05,
1869
+ "phase": "full_vq",
1870
+ "val_recon_loss": 0.2471015052153514,
1871
+ "val_utilization": 0.182861328125,
1872
+ "time_sec": 47.657753705978394
1873
+ },
1874
+ {
1875
+ "epoch": 144,
1876
+ "loss": 0.3041202864226173,
1877
+ "recon_loss": 0.24207541667947582,
1878
+ "commit_loss": 0.031022435127227913,
1879
+ "embed_loss": 0.031022435127227913,
1880
+ "codebook_utilization": 0.220703125,
1881
+ "lr": 1.837773122023114e-05,
1882
+ "phase": "full_vq",
1883
+ "val_recon_loss": 0.24735069504150978,
1884
+ "val_utilization": 0.183837890625,
1885
+ "time_sec": 45.71464681625366
1886
+ },
1887
+ {
1888
+ "epoch": 145,
1889
+ "loss": 0.3045578429511949,
1890
+ "recon_loss": 0.24211594959100088,
1891
+ "commit_loss": 0.031220946753141927,
1892
+ "embed_loss": 0.031220946753141927,
1893
+ "codebook_utilization": 0.2197265625,
1894
+ "lr": 1.775788936319148e-05,
1895
+ "phase": "full_vq",
1896
+ "val_recon_loss": 0.24681614568600288,
1897
+ "val_utilization": 0.1845703125,
1898
+ "time_sec": 46.50504207611084
1899
+ },
1900
+ {
1901
+ "epoch": 146,
1902
+ "loss": 0.30445228838453103,
1903
+ "recon_loss": 0.24190833580260182,
1904
+ "commit_loss": 0.03127197647357688,
1905
+ "embed_loss": 0.03127197647357688,
1906
+ "codebook_utilization": 0.220947265625,
1907
+ "lr": 1.714641594781347e-05,
1908
+ "phase": "full_vq",
1909
+ "val_recon_loss": 0.24688133597373962,
1910
+ "val_utilization": 0.181884765625,
1911
+ "time_sec": 46.81335687637329
1912
+ },
1913
+ {
1914
+ "epoch": 147,
1915
+ "loss": 0.30401302085203286,
1916
+ "recon_loss": 0.24176964485177807,
1917
+ "commit_loss": 0.031121688621009096,
1918
+ "embed_loss": 0.031121688621009096,
1919
+ "codebook_utilization": 0.2216796875,
1920
+ "lr": 1.654346968205709e-05,
1921
+ "phase": "full_vq",
1922
+ "val_recon_loss": 0.2471463428093837,
1923
+ "val_utilization": 0.184326171875,
1924
+ "time_sec": 46.61048150062561
1925
+ },
1926
+ {
1927
+ "epoch": 148,
1928
+ "loss": 0.30409211504693123,
1929
+ "recon_loss": 0.24158454204306884,
1930
+ "commit_loss": 0.031253786684543476,
1931
+ "embed_loss": 0.031253786684543476,
1932
+ "codebook_utilization": 0.220947265625,
1933
+ "lr": 1.5949207060660148e-05,
1934
+ "phase": "full_vq",
1935
+ "val_recon_loss": 0.24736738892701957,
1936
+ "val_utilization": 0.18310546875,
1937
+ "time_sec": 47.3840548992157
1938
+ },
1939
+ {
1940
+ "epoch": 149,
1941
+ "loss": 0.3043592572212219,
1942
+ "recon_loss": 0.24159174865367367,
1943
+ "commit_loss": 0.0313837544663864,
1944
+ "embed_loss": 0.0313837544663864,
1945
+ "codebook_utilization": 0.220947265625,
1946
+ "lr": 1.536378232452003e-05,
1947
+ "phase": "full_vq",
1948
+ "val_recon_loss": 0.24695723217267257,
1949
+ "val_utilization": 0.18310546875,
1950
+ "time_sec": 45.059221506118774
1951
+ },
1952
+ {
1953
+ "epoch": 150,
1954
+ "loss": 0.30416933753911185,
1955
+ "recon_loss": 0.24157329341944525,
1956
+ "commit_loss": 0.03129802100068214,
1957
+ "embed_loss": 0.03129802100068214,
1958
+ "codebook_utilization": 0.220458984375,
1959
+ "lr": 1.478734742066054e-05,
1960
+ "phase": "full_vq",
1961
+ "val_recon_loss": 0.2472969373831382,
1962
+ "val_utilization": 0.1845703125,
1963
+ "time_sec": 46.76996731758118
1964
+ },
1965
+ {
1966
+ "epoch": 151,
1967
+ "loss": 0.3044133408396852,
1968
+ "recon_loss": 0.24150650407753738,
1969
+ "commit_loss": 0.031453419586314875,
1970
+ "embed_loss": 0.031453419586314875,
1971
+ "codebook_utilization": 0.220703125,
1972
+ "lr": 1.4220051962793949e-05,
1973
+ "phase": "full_vq",
1974
+ "val_recon_loss": 0.24714479194237635,
1975
+ "val_utilization": 0.183837890625,
1976
+ "time_sec": 47.58507513999939
1977
+ },
1978
+ {
1979
+ "epoch": 152,
1980
+ "loss": 0.3045247682169372,
1981
+ "recon_loss": 0.24158037380844938,
1982
+ "commit_loss": 0.031472196948586724,
1983
+ "embed_loss": 0.031472196948586724,
1984
+ "codebook_utilization": 0.22021484375,
1985
+ "lr": 1.3662043192488835e-05,
1986
+ "phase": "full_vq",
1987
+ "val_recon_loss": 0.24677677337939924,
1988
+ "val_utilization": 0.1865234375,
1989
+ "time_sec": 47.2166223526001
1990
+ },
1991
+ {
1992
+ "epoch": 153,
1993
+ "loss": 0.3043605843011071,
1994
+ "recon_loss": 0.24142262953169205,
1995
+ "commit_loss": 0.031468977056005425,
1996
+ "embed_loss": 0.031468977056005425,
1997
+ "codebook_utilization": 0.219970703125,
1998
+ "lr": 1.3113465940953492e-05,
1999
+ "phase": "full_vq",
2000
+ "val_recon_loss": 0.24657305845847496,
2001
+ "val_utilization": 0.181884765625,
2002
+ "time_sec": 47.24207806587219
2003
+ },
2004
+ {
2005
+ "epoch": 154,
2006
+ "loss": 0.30447067641744424,
2007
+ "recon_loss": 0.24142640536906673,
2008
+ "commit_loss": 0.03152213526853159,
2009
+ "embed_loss": 0.03152213526853159,
2010
+ "codebook_utilization": 0.220703125,
2011
+ "lr": 1.2574462591444937e-05,
2012
+ "phase": "full_vq",
2013
+ "val_recon_loss": 0.24718056389918694,
2014
+ "val_utilization": 0.184814453125,
2015
+ "time_sec": 47.69112420082092
2016
+ },
2017
+ {
2018
+ "epoch": 155,
2019
+ "loss": 0.30477023416874455,
2020
+ "recon_loss": 0.24140178163846335,
2021
+ "commit_loss": 0.031684226776454966,
2022
+ "embed_loss": 0.031684226776454966,
2023
+ "codebook_utilization": 0.219970703125,
2024
+ "lr": 1.2045173042313425e-05,
2025
+ "phase": "full_vq",
2026
+ "val_recon_loss": 0.24684409682567304,
2027
+ "val_utilization": 0.182861328125,
2028
+ "time_sec": 48.42423105239868
2029
+ },
2030
+ {
2031
+ "epoch": 156,
2032
+ "loss": 0.3046512428451987,
2033
+ "recon_loss": 0.241388949693418,
2034
+ "commit_loss": 0.03163114675850261,
2035
+ "embed_loss": 0.03163114675850261,
2036
+ "codebook_utilization": 0.219970703125,
2037
+ "lr": 1.1525734670691714e-05,
2038
+ "phase": "full_vq",
2039
+ "val_recon_loss": 0.2468604938342021,
2040
+ "val_utilization": 0.184814453125,
2041
+ "time_sec": 47.058255434036255
2042
+ },
2043
+ {
2044
+ "epoch": 157,
2045
+ "loss": 0.30505996416596803,
2046
+ "recon_loss": 0.24163535120440463,
2047
+ "commit_loss": 0.03171230585989999,
2048
+ "embed_loss": 0.03171230585989999,
2049
+ "codebook_utilization": 0.220458984375,
2050
+ "lr": 1.1016282296838872e-05,
2051
+ "phase": "full_vq",
2052
+ "val_recon_loss": 0.24692615981285387,
2053
+ "val_utilization": 0.1845703125,
2054
+ "time_sec": 48.85962629318237
2055
+ },
2056
+ {
2057
+ "epoch": 158,
2058
+ "loss": 0.3047319282503689,
2059
+ "recon_loss": 0.24128261150098315,
2060
+ "commit_loss": 0.03172465669466,
2061
+ "embed_loss": 0.03172465669466,
2062
+ "codebook_utilization": 0.2197265625,
2063
+ "lr": 1.051694814914775e-05,
2064
+ "phase": "full_vq",
2065
+ "val_recon_loss": 0.24680422017207512,
2066
+ "val_utilization": 0.186279296875,
2067
+ "time_sec": 47.58470416069031
2068
+ },
2069
+ {
2070
+ "epoch": 159,
2071
+ "loss": 0.305074086376265,
2072
+ "recon_loss": 0.24131639593956516,
2073
+ "commit_loss": 0.03187884463399064,
2074
+ "embed_loss": 0.03187884463399064,
2075
+ "codebook_utilization": 0.22021484375,
2076
+ "lr": 1.002786182982495e-05,
2077
+ "phase": "full_vq",
2078
+ "val_recon_loss": 0.24687475653795096,
2079
+ "val_utilization": 0.18359375,
2080
+ "time_sec": 46.899904012680054
2081
+ },
2082
+ {
2083
+ "epoch": 160,
2084
+ "loss": 0.30493279473454343,
2085
+ "recon_loss": 0.2412821378778009,
2086
+ "commit_loss": 0.0318253295240449,
2087
+ "embed_loss": 0.0318253295240449,
2088
+ "codebook_utilization": 0.220458984375,
2089
+ "lr": 9.549150281252631e-06,
2090
+ "phase": "full_vq",
2091
+ "val_recon_loss": 0.24676664402851692,
2092
+ "val_utilization": 0.184814453125,
2093
+ "time_sec": 47.877979040145874
2094
+ },
2095
+ {
2096
+ "epoch": 161,
2097
+ "loss": 0.3051188383616653,
2098
+ "recon_loss": 0.24118808379360274,
2099
+ "commit_loss": 0.03196537874492945,
2100
+ "embed_loss": 0.03196537874492945,
2101
+ "codebook_utilization": 0.22021484375,
2102
+ "lr": 9.080937753040662e-06,
2103
+ "phase": "full_vq",
2104
+ "val_recon_loss": 0.24653198168827936,
2105
+ "val_utilization": 0.184326171875,
2106
+ "time_sec": 49.28059220314026
2107
+ },
2108
+ {
2109
+ "epoch": 162,
2110
+ "loss": 0.3049393664388096,
2111
+ "recon_loss": 0.24099352634420582,
2112
+ "commit_loss": 0.03197291946294261,
2113
+ "embed_loss": 0.03197291946294261,
2114
+ "codebook_utilization": 0.22021484375,
2115
+ "lr": 8.62334576977751e-06,
2116
+ "phase": "full_vq",
2117
+ "val_recon_loss": 0.24671129996959978,
2118
+ "val_utilization": 0.18505859375,
2119
+ "time_sec": 48.106534242630005
2120
+ },
2121
+ {
2122
+ "epoch": 163,
2123
+ "loss": 0.30524564782778424,
2124
+ "recon_loss": 0.24120154071087932,
2125
+ "commit_loss": 0.0320220530471381,
2126
+ "embed_loss": 0.0320220530471381,
2127
+ "codebook_utilization": 0.2197265625,
2128
+ "lr": 8.176493099488661e-06,
2129
+ "phase": "full_vq",
2130
+ "val_recon_loss": 0.24669795540662912,
2131
+ "val_utilization": 0.18359375,
2132
+ "time_sec": 47.48547387123108
2133
+ },
2134
+ {
2135
+ "epoch": 164,
2136
+ "loss": 0.30536650734789234,
2137
+ "recon_loss": 0.24124695419096479,
2138
+ "commit_loss": 0.03205977610367186,
2139
+ "embed_loss": 0.03205977610367186,
2140
+ "codebook_utilization": 0.2197265625,
2141
+ "lr": 7.740495722810268e-06,
2142
+ "phase": "full_vq",
2143
+ "val_recon_loss": 0.2465652869297908,
2144
+ "val_utilization": 0.184326171875,
2145
+ "time_sec": 45.63185739517212
2146
+ },
2147
+ {
2148
+ "epoch": 165,
2149
+ "loss": 0.30516086723290237,
2150
+ "recon_loss": 0.24114423610416114,
2151
+ "commit_loss": 0.03200831596611762,
2152
+ "embed_loss": 0.03200831596611762,
2153
+ "codebook_utilization": 0.2197265625,
2154
+ "lr": 7.315466802886386e-06,
2155
+ "phase": "full_vq",
2156
+ "val_recon_loss": 0.24668536621790665,
2157
+ "val_utilization": 0.183837890625,
2158
+ "time_sec": 46.74771308898926
2159
+ },
2160
+ {
2161
+ "epoch": 166,
2162
+ "loss": 0.30522307811998856,
2163
+ "recon_loss": 0.24109522384755752,
2164
+ "commit_loss": 0.032063927464917596,
2165
+ "embed_loss": 0.032063927464917596,
2166
+ "codebook_utilization": 0.220458984375,
2167
+ "lr": 6.901516655997545e-06,
2168
+ "phase": "full_vq",
2169
+ "val_recon_loss": 0.24656208891134995,
2170
+ "val_utilization": 0.184326171875,
2171
+ "time_sec": 47.7575261592865
2172
+ },
2173
+ {
2174
+ "epoch": 167,
2175
+ "loss": 0.3054830934487137,
2176
+ "recon_loss": 0.24122415833613453,
2177
+ "commit_loss": 0.032129466387571075,
2178
+ "embed_loss": 0.032129466387571075,
2179
+ "codebook_utilization": 0.219970703125,
2180
+ "lr": 6.4987527229280395e-06,
2181
+ "phase": "full_vq",
2182
+ "val_recon_loss": 0.24661564941589648,
2183
+ "val_utilization": 0.184814453125,
2184
+ "time_sec": 47.754785776138306
2185
+ },
2186
+ {
2187
+ "epoch": 168,
2188
+ "loss": 0.3055601499828638,
2189
+ "recon_loss": 0.24107691265788733,
2190
+ "commit_loss": 0.03224161910075767,
2191
+ "embed_loss": 0.03224161910075767,
2192
+ "codebook_utilization": 0.219482421875,
2193
+ "lr": 6.107279541079777e-06,
2194
+ "phase": "full_vq",
2195
+ "val_recon_loss": 0.246795192360878,
2196
+ "val_utilization": 0.18505859375,
2197
+ "time_sec": 45.297162771224976
2198
+ },
2199
+ {
2200
+ "epoch": 169,
2201
+ "loss": 0.3054829397622277,
2202
+ "recon_loss": 0.24103103519654742,
2203
+ "commit_loss": 0.03222595202718295,
2204
+ "embed_loss": 0.03222595202718295,
2205
+ "codebook_utilization": 0.2197265625,
2206
+ "lr": 5.727198717339518e-06,
2207
+ "phase": "full_vq",
2208
+ "val_recon_loss": 0.24653366322700793,
2209
+ "val_utilization": 0.1845703125,
2210
+ "time_sec": 46.21055722236633
2211
+ },
2212
+ {
2213
+ "epoch": 170,
2214
+ "loss": 0.30541764988618736,
2215
+ "recon_loss": 0.24095032320303075,
2216
+ "commit_loss": 0.03223366440072948,
2217
+ "embed_loss": 0.03223366440072948,
2218
+ "codebook_utilization": 0.21875,
2219
+ "lr": 5.358608901706794e-06,
2220
+ "phase": "full_vq",
2221
+ "val_recon_loss": 0.24648518974964434,
2222
+ "val_utilization": 0.184326171875,
2223
+ "time_sec": 47.754228353500366
2224
+ },
2225
+ {
2226
+ "epoch": 171,
2227
+ "loss": 0.3055083266660279,
2228
+ "recon_loss": 0.24096162821732314,
2229
+ "commit_loss": 0.03227335130613224,
2230
+ "embed_loss": 0.03227335130613224,
2231
+ "codebook_utilization": 0.218994140625,
2232
+ "lr": 5.001605761689385e-06,
2233
+ "phase": "full_vq",
2234
+ "val_recon_loss": 0.2468345589362658,
2235
+ "val_utilization": 0.18603515625,
2236
+ "time_sec": 46.85165524482727
2237
+ },
2238
+ {
2239
+ "epoch": 172,
2240
+ "loss": 0.30541246252901416,
2241
+ "recon_loss": 0.24084528199597902,
2242
+ "commit_loss": 0.03228359169089327,
2243
+ "embed_loss": 0.03228359169089327,
2244
+ "codebook_utilization": 0.218994140625,
2245
+ "lr": 4.656281957472728e-06,
2246
+ "phase": "full_vq",
2247
+ "val_recon_loss": 0.24658531638292167,
2248
+ "val_utilization": 0.1865234375,
2249
+ "time_sec": 45.791006088256836
2250
+ },
2251
+ {
2252
+ "epoch": 173,
2253
+ "loss": 0.3054757223409765,
2254
+ "recon_loss": 0.24092271953236824,
2255
+ "commit_loss": 0.0322765015869164,
2256
+ "embed_loss": 0.0322765015869164,
2257
+ "codebook_utilization": 0.21923828125,
2258
+ "lr": 4.322727117869949e-06,
2259
+ "phase": "full_vq",
2260
+ "val_recon_loss": 0.2463932312451876,
2261
+ "val_utilization": 0.1845703125,
2262
+ "time_sec": 46.77355599403381
2263
+ },
2264
+ {
2265
+ "epoch": 174,
2266
+ "loss": 0.3055362193023457,
2267
+ "recon_loss": 0.24084466987965153,
2268
+ "commit_loss": 0.03234577493048182,
2269
+ "embed_loss": 0.03234577493048182,
2270
+ "codebook_utilization": 0.21923828125,
2271
+ "lr": 4.0010278170587865e-06,
2272
+ "phase": "full_vq",
2273
+ "val_recon_loss": 0.24642541660712317,
2274
+ "val_utilization": 0.18408203125,
2275
+ "time_sec": 47.291191816329956
2276
+ },
2277
+ {
2278
+ "epoch": 175,
2279
+ "loss": 0.30543482829542723,
2280
+ "recon_loss": 0.24080353390936757,
2281
+ "commit_loss": 0.03231564573213166,
2282
+ "embed_loss": 0.03231564573213166,
2283
+ "codebook_utilization": 0.2197265625,
2284
+ "lr": 3.6912675521111804e-06,
2285
+ "phase": "full_vq",
2286
+ "val_recon_loss": 0.24655043276456687,
2287
+ "val_utilization": 0.1845703125,
2288
+ "time_sec": 46.936854124069214
2289
+ },
2290
+ {
2291
+ "epoch": 176,
2292
+ "loss": 0.3055465192187066,
2293
+ "recon_loss": 0.24077600006963693,
2294
+ "commit_loss": 0.032385260670208464,
2295
+ "embed_loss": 0.032385260670208464,
2296
+ "codebook_utilization": 0.21875,
2297
+ "lr": 3.393526721321614e-06,
2298
+ "phase": "full_vq",
2299
+ "val_recon_loss": 0.2467047904546444,
2300
+ "val_utilization": 0.185791015625,
2301
+ "time_sec": 45.46668100357056
2302
+ },
2303
+ {
2304
+ "epoch": 177,
2305
+ "loss": 0.3057172917852215,
2306
+ "recon_loss": 0.2408869146716361,
2307
+ "commit_loss": 0.0324151891046295,
2308
+ "embed_loss": 0.0324151891046295,
2309
+ "codebook_utilization": 0.2197265625,
2310
+ "lr": 3.1078826033397826e-06,
2311
+ "phase": "full_vq",
2312
+ "val_recon_loss": 0.24667876385725462,
2313
+ "val_utilization": 0.185546875,
2314
+ "time_sec": 45.404324531555176
2315
+ },
2316
+ {
2317
+ "epoch": 178,
2318
+ "loss": 0.30570807351785545,
2319
+ "recon_loss": 0.24085690753132688,
2320
+ "commit_loss": 0.032425583175876564,
2321
+ "embed_loss": 0.032425583175876564,
2322
+ "codebook_utilization": 0.21875,
2323
+ "lr": 2.834409337112846e-06,
2324
+ "phase": "full_vq",
2325
+ "val_recon_loss": 0.24646423413203314,
2326
+ "val_utilization": 0.185546875,
2327
+ "time_sec": 47.09632229804993
2328
+ },
2329
+ {
2330
+ "epoch": 179,
2331
+ "loss": 0.3056108349678563,
2332
+ "recon_loss": 0.2407783240664239,
2333
+ "commit_loss": 0.032416255377671295,
2334
+ "embed_loss": 0.032416255377671295,
2335
+ "codebook_utilization": 0.218994140625,
2336
+ "lr": 2.57317790264273e-06,
2337
+ "phase": "full_vq",
2338
+ "val_recon_loss": 0.24667887962781465,
2339
+ "val_utilization": 0.185546875,
2340
+ "time_sec": 47.75557231903076
2341
+ },
2342
+ {
2343
+ "epoch": 180,
2344
+ "loss": 0.30572008969737036,
2345
+ "recon_loss": 0.24083752637984707,
2346
+ "commit_loss": 0.03244128297356998,
2347
+ "embed_loss": 0.03244128297356998,
2348
+ "codebook_utilization": 0.218994140625,
2349
+ "lr": 2.324256102563187e-06,
2350
+ "phase": "full_vq",
2351
+ "val_recon_loss": 0.24677492678165436,
2352
+ "val_utilization": 0.186767578125,
2353
+ "time_sec": 46.7584228515625
2354
+ },
2355
+ {
2356
+ "epoch": 181,
2357
+ "loss": 0.3058477476531384,
2358
+ "recon_loss": 0.24080230061914407,
2359
+ "commit_loss": 0.032522723115250174,
2360
+ "embed_loss": 0.032522723115250174,
2361
+ "codebook_utilization": 0.21923828125,
2362
+ "lr": 2.0877085445416877e-06,
2363
+ "phase": "full_vq",
2364
+ "val_recon_loss": 0.24674974038050726,
2365
+ "val_utilization": 0.185791015625,
2366
+ "time_sec": 46.70277667045593
2367
+ },
2368
+ {
2369
+ "epoch": 182,
2370
+ "loss": 0.30580803576637716,
2371
+ "recon_loss": 0.24082145766884672,
2372
+ "commit_loss": 0.032493290107916385,
2373
+ "embed_loss": 0.032493290107916385,
2374
+ "codebook_utilization": 0.21923828125,
2375
+ "lr": 1.8635966245104596e-06,
2376
+ "phase": "full_vq",
2377
+ "val_recon_loss": 0.24670579456366026,
2378
+ "val_utilization": 0.185791015625,
2379
+ "time_sec": 47.609474420547485
2380
+ },
2381
+ {
2382
+ "epoch": 183,
2383
+ "loss": 0.30572647265359465,
2384
+ "recon_loss": 0.24077031104003682,
2385
+ "commit_loss": 0.03247807930935832,
2386
+ "embed_loss": 0.03247807930935832,
2387
+ "codebook_utilization": 0.21923828125,
2388
+ "lr": 1.651978510731188e-06,
2389
+ "phase": "full_vq",
2390
+ "val_recon_loss": 0.24664189723821786,
2391
+ "val_utilization": 0.18603515625,
2392
+ "time_sec": 44.83900165557861
2393
+ },
2394
+ {
2395
+ "epoch": 184,
2396
+ "loss": 0.30577124509156917,
2397
+ "recon_loss": 0.24073131090285732,
2398
+ "commit_loss": 0.0325199676787152,
2399
+ "embed_loss": 0.0325199676787152,
2400
+ "codebook_utilization": 0.21923828125,
2401
+ "lr": 1.4529091286973986e-06,
2402
+ "phase": "full_vq",
2403
+ "val_recon_loss": 0.24661937470619494,
2404
+ "val_utilization": 0.185302734375,
2405
+ "time_sec": 43.470951795578
2406
+ },
2407
+ {
2408
+ "epoch": 185,
2409
+ "loss": 0.30575791351935444,
2410
+ "recon_loss": 0.24074466117456847,
2411
+ "commit_loss": 0.03250662617239298,
2412
+ "embed_loss": 0.03250662617239298,
2413
+ "codebook_utilization": 0.21923828125,
2414
+ "lr": 1.2664401468786106e-06,
2415
+ "phase": "full_vq",
2416
+ "val_recon_loss": 0.24674029992176935,
2417
+ "val_utilization": 0.18603515625,
2418
+ "time_sec": 43.349894285202026
2419
+ },
2420
+ {
2421
+ "epoch": 186,
2422
+ "loss": 0.3058706136310802,
2423
+ "recon_loss": 0.24078886269354352,
2424
+ "commit_loss": 0.032540876235739855,
2425
+ "embed_loss": 0.032540876235739855,
2426
+ "codebook_utilization": 0.21923828125,
2427
+ "lr": 1.0926199633097149e-06,
2428
+ "phase": "full_vq",
2429
+ "val_recon_loss": 0.2466356662603525,
2430
+ "val_utilization": 0.185546875,
2431
+ "time_sec": 44.53398299217224
2432
+ },
2433
+ {
2434
+ "epoch": 187,
2435
+ "loss": 0.30575790358524696,
2436
+ "recon_loss": 0.24070138791028192,
2437
+ "commit_loss": 0.0325282594444705,
2438
+ "embed_loss": 0.0325282594444705,
2439
+ "codebook_utilization": 0.21923828125,
2440
+ "lr": 9.314936930293277e-07,
2441
+ "phase": "full_vq",
2442
+ "val_recon_loss": 0.24666727506197417,
2443
+ "val_utilization": 0.185791015625,
2444
+ "time_sec": 45.00711798667908
2445
+ },
2446
+ {
2447
+ "epoch": 188,
2448
+ "loss": 0.3058345446399614,
2449
+ "recon_loss": 0.24071199286217784,
2450
+ "commit_loss": 0.032561275560189694,
2451
+ "embed_loss": 0.032561275560189694,
2452
+ "codebook_utilization": 0.21923828125,
2453
+ "lr": 7.83103156370118e-07,
2454
+ "phase": "full_vq",
2455
+ "val_recon_loss": 0.24662736172859484,
2456
+ "val_utilization": 0.185791015625,
2457
+ "time_sec": 47.04360866546631
2458
+ },
2459
+ {
2460
+ "epoch": 189,
2461
+ "loss": 0.3057972625190136,
2462
+ "recon_loss": 0.24071210242953955,
2463
+ "commit_loss": 0.03254257909515325,
2464
+ "embed_loss": 0.03254257909515325,
2465
+ "codebook_utilization": 0.21923828125,
2466
+ "lr": 6.474868681043573e-07,
2467
+ "phase": "full_vq",
2468
+ "val_recon_loss": 0.24643382544700915,
2469
+ "val_utilization": 0.185791015625,
2470
+ "time_sec": 46.66129493713379
2471
+ },
2472
+ {
2473
+ "epoch": 190,
2474
+ "loss": 0.3058857058777529,
2475
+ "recon_loss": 0.24079456662430482,
2476
+ "commit_loss": 0.032545569188454575,
2477
+ "embed_loss": 0.032545569188454575,
2478
+ "codebook_utilization": 0.2197265625,
2479
+ "lr": 5.246800274474491e-07,
2480
+ "phase": "full_vq",
2481
+ "val_recon_loss": 0.24659072091946235,
2482
+ "val_utilization": 0.18603515625,
2483
+ "time_sec": 45.12866711616516
2484
+ },
2485
+ {
2486
+ "epoch": 191,
2487
+ "loss": 0.3059348025742699,
2488
+ "recon_loss": 0.2407972374383141,
2489
+ "commit_loss": 0.032568780705332756,
2490
+ "embed_loss": 0.032568780705332756,
2491
+ "codebook_utilization": 0.2197265625,
2492
+ "lr": 4.1471450892189815e-07,
2493
+ "phase": "full_vq",
2494
+ "val_recon_loss": 0.246567142697481,
2495
+ "val_utilization": 0.185546875,
2496
+ "time_sec": 44.866029262542725
2497
+ },
2498
+ {
2499
+ "epoch": 192,
2500
+ "loss": 0.3058386427514693,
2501
+ "recon_loss": 0.24068779951217129,
2502
+ "commit_loss": 0.03257542169269394,
2503
+ "embed_loss": 0.03257542169269394,
2504
+ "codebook_utilization": 0.2197265625,
2505
+ "lr": 3.1761885408435033e-07,
2506
+ "phase": "full_vq",
2507
+ "val_recon_loss": 0.24653759827980629,
2508
+ "val_utilization": 0.185546875,
2509
+ "time_sec": 46.77774691581726
2510
+ },
2511
+ {
2512
+ "epoch": 193,
2513
+ "loss": 0.3059851235034419,
2514
+ "recon_loss": 0.24080172940796496,
2515
+ "commit_loss": 0.03259169774166509,
2516
+ "embed_loss": 0.03259169774166509,
2517
+ "codebook_utilization": 0.2197265625,
2518
+ "lr": 2.3341826411756845e-07,
2519
+ "phase": "full_vq",
2520
+ "val_recon_loss": 0.2464261524952375,
2521
+ "val_utilization": 0.185791015625,
2522
+ "time_sec": 47.40852236747742
2523
+ },
2524
+ {
2525
+ "epoch": 194,
2526
+ "loss": 0.3058479352324617,
2527
+ "recon_loss": 0.24067929620836295,
2528
+ "commit_loss": 0.03258431885464519,
2529
+ "embed_loss": 0.03258431885464519,
2530
+ "codebook_utilization": 0.2197265625,
2531
+ "lr": 1.6213459328950342e-07,
2532
+ "phase": "full_vq",
2533
+ "val_recon_loss": 0.2464728607581212,
2534
+ "val_utilization": 0.185546875,
2535
+ "time_sec": 48.14972186088562
2536
+ },
2537
+ {
2538
+ "epoch": 195,
2539
+ "loss": 0.30596720120486093,
2540
+ "recon_loss": 0.240788919376392,
2541
+ "commit_loss": 0.03258914164468354,
2542
+ "embed_loss": 0.03258914164468354,
2543
+ "codebook_utilization": 0.2197265625,
2544
+ "lr": 1.037863432809926e-07,
2545
+ "phase": "full_vq",
2546
+ "val_recon_loss": 0.2464477545939959,
2547
+ "val_utilization": 0.185791015625,
2548
+ "time_sec": 47.88664484024048
2549
+ },
2550
+ {
2551
+ "epoch": 196,
2552
+ "loss": 0.30592520973261667,
2553
+ "recon_loss": 0.24075032478454067,
2554
+ "commit_loss": 0.032587441816633825,
2555
+ "embed_loss": 0.032587441816633825,
2556
+ "codebook_utilization": 0.2197265625,
2557
+ "lr": 5.8388658383673425e-08,
2558
+ "phase": "full_vq",
2559
+ "val_recon_loss": 0.24659524170251992,
2560
+ "val_utilization": 0.185791015625,
2561
+ "time_sec": 47.65922021865845
2562
+ },
2563
+ {
2564
+ "epoch": 197,
2565
+ "loss": 0.30599646007313447,
2566
+ "recon_loss": 0.24082232807196824,
2567
+ "commit_loss": 0.03258706614667294,
2568
+ "embed_loss": 0.03258706614667294,
2569
+ "codebook_utilization": 0.2197265625,
2570
+ "lr": 2.5953321569255322e-08,
2571
+ "phase": "full_vq",
2572
+ "val_recon_loss": 0.2464762547841439,
2573
+ "val_utilization": 0.185791015625,
2574
+ "time_sec": 47.21767497062683
2575
+ },
2576
+ {
2577
+ "epoch": 198,
2578
+ "loss": 0.3059417284002491,
2579
+ "recon_loss": 0.24077224263957903,
2580
+ "commit_loss": 0.0325847425151105,
2581
+ "embed_loss": 0.0325847425151105,
2582
+ "codebook_utilization": 0.2197265625,
2583
+ "lr": 6.4887514312661454e-09,
2584
+ "phase": "full_vq",
2585
+ "val_recon_loss": 0.24647761766727155,
2586
+ "val_utilization": 0.185791015625,
2587
+ "time_sec": 46.83630871772766
2588
+ },
2589
+ {
2590
+ "epoch": 199,
2591
+ "loss": 0.30583253619717615,
2592
+ "recon_loss": 0.24067074060440063,
2593
+ "commit_loss": 0.03258089743116323,
2594
+ "embed_loss": 0.03258089743116323,
2595
+ "codebook_utilization": 0.2197265625,
2596
+ "lr": 0.0,
2597
+ "phase": "full_vq",
2598
+ "val_recon_loss": 0.2466405824973033,
2599
+ "val_utilization": 0.18603515625,
2600
+ "time_sec": 46.01119041442871
2601
+ }
2602
+ ]