added saving tests to cpu
This commit is contained in:
parent
fbc1bbd161
commit
a374a7ea00
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue