added saving tests to cpu

This commit is contained in:
William Falcon 2019-07-26 12:33:35 -04:00
parent fbc1bbd161
commit a374a7ea00
1 changed files with 2 additions and 1 deletions

View File

@ -27,6 +27,8 @@ np.random.seed(SEED)
def test_loading_meta_tags():
hparams = get_hparams()
save_dir = init_save_dir()
# save tags
exp = get_exp(False)
exp.tag({'some_str':'a_str', 'an_int': 1, 'a_float': 2.0})
@ -41,7 +43,6 @@ def test_loading_meta_tags():
clear_save_dir()
def test_dp_output_reduce():
# test identity when we have a single gpu