@arya47
outputs:
Epoch 1
-------------------------------
Train Auccuracy: 73.8 , Train loss: 1.0836 

Test Auccuracy: 90.2 , Test loss: 0.3438 

Epoch 2
-------------------------------
Train Auccuracy: 91.2 , Train loss: 0.3006 

Test Auccuracy: 92.7 , Test loss: 0.2449 

Epoch 3
-------------------------------
Train Auccuracy: 93.5 , Train loss: 0.2227 

Test Auccuracy: 94.7 , Test loss: 0.1786 

Epoch 4
-------------------------------
Train Auccuracy: 94.8 , Train loss: 0.1761 

Test Auccuracy: 95.8 , Test loss: 0.1449 

Epoch 5
-------------------------------
Train Auccuracy: 95.7 , Train loss: 0.1468
Test Auccuracy: 96.3 , Test loss: 0.1235 

Done!

Could you tell me how to do parameter dataloader?