test_mnist_amp.py 3.5 KB