mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
added saving tests to cpu
This commit is contained in:
1 parent
fbc1bbd161
commit
a374a7ea00
1 file changed
+2
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user