testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1562.58it/s]epoch: 0
avg loss: 0.6662371538425887
accuracy: 0.6769724323932329
test accuracy: 0.6496585735963581

testing:  97%|█████████▋| 5120/5272 [00:04<00:00, 1485.70it/s]epoch: 1
avg loss: 0.6602801650684239
accuracy: 0.6871561819054558
test accuracy: 0.6574355083459787

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1509.15it/s]epoch: 2
avg loss: 0.6611704620506444
accuracy: 0.6899844256662258
test accuracy: 0.6566767830045523

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1471.36it/s]epoch: 3
avg loss: 0.6616791084902397
accuracy: 0.6911412775097642
test accuracy: 0.6638846737481032

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1579.47it/s]epoch: 4
avg loss: 0.6610813111163456
accuracy: 0.6913588347221311
test accuracy: 0.6403641881638846

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1561.11it/s]epoch: 5
avg loss: 0.6612948830510013
accuracy: 0.6919113609757613
test accuracy: 0.6553490136570561

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1496.72it/s]epoch: 6
avg loss: 0.662789237758215
accuracy: 0.6916558175834574
test accuracy: 0.6688163884673748   <--- this is the best we managed to get

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1531.05it/s]epoch: 7
avg loss: 0.6635299078017594
accuracy: 0.6916730840288833
test accuracy: 0.6525037936267072

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1539.94it/s]epoch: 8
avg loss: 0.6665739091241887
accuracy: 0.6917179767869908
test accuracy: 0.6676783004552352