INFO:root:Number of parameters: 61913602
INFO:root:Starting smoke training from random initialization
ERROR:root:
  0%|                                                                                                                                                             | 0/3 [00:00<?, ?it/s]
ERROR:root:
 33%|█████████████████████████████████████████████████▋                                                                                                   | 1/3 [01:13<02:26, 73.40s/it]
ERROR:root:
 67%|███████████████████████████████████████████████████████████████████████████████████████████████████▎                                                 | 2/3 [01:13<00:30, 30.31s/it]
ERROR:root:
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3/3 [01:13<00:00, 16.53s/it]
ERROR:root:
ERROR:root:
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3/3 [01:13<00:00, 24.56s/it]
ERROR:root:

INFO:root:Epoch 1/1, Loss: 0.037418019451855244
INFO:root:
Best validation loss: 0.037418019451855244
INFO:root:
Saving best model for epoch: 1

INFO:root:Saved new best model: ./smoke_train_data/0_best_model.pth
INFO:root:Test loss: 0.037418019451855244
INFO:root:Test MAE: 7.171685719208052e-05