swh:1:snp:d2bcff616bbf538fe8ce2a9c384200307730292a
Tip revision: ff05f7e8ae2fc51529f05f4f7a849ea545d4f8cf authored by Felix Leibfried on 21 January 2020, 13:41:47 UTC
minor change to optimizer
minor change to optimizer
Tip revision: ff05f7e
test_examples.py
import sys
import pytest
sys.path.append('./examples')
# from train_mnist import ex
@pytest.mark.examples
def test_train_mnist():
"""
Just check whether the train_mist script still runs...
NB: This script will fail if you test twice within the same minute, due to
the name where output files are dumped being the same. Not a problem IRL.
"""
# TODO: Run this test! Currently this doesn't run on CI, due to a bug in some metadata that TF should be giving
# as part of importing itself. This makes it not work with sacred. A PR has been merged into TF, but this
# hasn't made it into a release yet. https://github.com/tensorflow/tensorflow/issues/30028
# ex.run(None, config_updates={"iterations": 2, "hz": {"short": 1}, "float_type": "float32", "jitter_level": 1e-4,
# "num_inducing_points": 100, "dataset": "mnist-small-subset-for-test"})