testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1496.45it/s]epoch: 0
avg loss: 0.5669572093742932
accuracy: 0.705209252052117
test accuracy: 0.7043400606980273

testing:  97%|█████████▋| 5120/5272 [00:03<00:00, 1500.00it/s]epoch: 1
avg loss: 0.5200198413747087
accuracy: 0.7267515945562351
test accuracy: 0.6817147192716236  <-- starts over-fitting