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_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)
