File size: 2,305 Bytes
de2c4f3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | 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
|