126 classes in this dataset
Ep: 0 lr: 0.01, loss_all: 5.239571, loss_c: 5.016991, loss_d: 2.225803, loss_u: 0.000000, loss_mme: 0.463399
