added saving tests to cpu

tests/test_models.py

@@ -27,8 +27,6 @@ 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})
@@ -39,8 +37,9 @@ def test_loading_meta_tags():
     tags_path = exp.get_data_path(exp.name, exp.version) + '/meta_tags.csv'
     tags = model_saving.load_hparams_from_tags_csv(tags_path)
 
-    pdb.set_trace()
-    assert len(tags) >=3
+    assert tags['batch_size'] == 32 and tags['hidden_dim'] == 1000
+
+    clear_save_dir()
 
 
 def test_dp_output_reduce():