python train.py
--------------------------------------------------------------------------------
DATALOADER:0 TEST RESULTS
{'Test/Accuracy': 0.9186999797821045, 'Test/loss': 0.28279516100883484}
--------------------------------------------------------------------------------