diff --git a/train.py b/train.py index e36829f..04ff3cf 100644 --- a/train.py +++ b/train.py @@ -1,5 +1,5 @@ '''Trains a simple convnet on the MNIST dataset. -Gets to 99.25% test accuracy after 12 epochs +Gets to 99.25% test accuracy after 12 epochs, (there is still a lot of margin for parameter tuning). 16 seconds per epoch on a GRID K520 GPU. '''