/pytorch-lightning-tutorial-mnist

Primary LanguagePython

How to run

python train.py

Expected output

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

References