Exemplo n.º 1
0
 def setUpClass(cls):
     download_mnist(os.path.dirname(__file__))
     cls.accuracy_tolerances = 3.0
     cls.generic_arguments = {
         "--batch-size": 4,
         "--batches-per-step": 1000,
         "--epochs": 10
     }
Exemplo n.º 2
0
def pytest_sessionstart(session):
    """Load the mnist data at the start of the session"""
    download_mnist(os.path.dirname(__file__))
 def setUpClass(cls):
     download_mnist(os.path.dirname(__file__))
     cls.cwd = os.path.dirname(__file__)
     cls.accuracy_tolerances = 3.0