def test_mnist(use_GPU): demos.mnist(model_args={"use_GPU": use_GPU}, run_args={ "minibatch_size": 7500, "max_epochs": 1 }) demos.profile("mnist", max_epochs=1, use_GPU=use_GPU)
def test_standard(): # just run through all the demos to make sure they run without crashing demos.xor() demos.crossentropy() demos.connections() demos.integrator(plots=False) demos.plant(plots=False) demos.profile("integrator", max_epochs=2)
def test_mnist(use_GPU): demos.mnist(model_args={"use_GPU": use_GPU}, run_args={"minibatch_size": 7500, "max_epochs": 1}) demos.profile("mnist", max_epochs=1, use_GPU=use_GPU)