added saving tests to cpu

2026-09-11 12:31:23 +08:00 · 2019-07-26 12:35:28 -04:00
parent a374a7ea00
commit 84edf35f33
1 changed files with 1 additions and 1 deletions
@@ -39,7 +39,7 @@ 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)

-    assert tags['batch_size'] == 32 and tags['hidden_dim'] == 1000
+    assert tags.batch_size == 32 and tags.hidden_dim == 1000

    clear_save_dir()