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 }
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