update test_pretrained

phoenix104104 · Dec 4, 2018 · 3dec0c0 · 3dec0c0
1 parent 563b152
commit 3dec0c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test_pretrained.py b/test_pretrained.py
@@ -42,7 +42,7 @@
 
 
     ### load model opts
-    opts_filename = os.path.join('pretrained_models', "ECCV18_model_opts.pth")
+    opts_filename = os.path.join('pretrained_models', "ECCV18_blind_consistency_opts.pth")
     print("Load %s" %opts_filename)
     with open(opts_filename, 'r') as f:
         model_opts = pickle.load(f)
@@ -54,7 +54,7 @@
 
 
     ### load trained model
-    model_filename = os.path.join('pretrained_models', "ECCV18_model.pth")
+    model_filename = os.path.join('pretrained_models', "ECCV18_blind_consistency.pth")
     print("Load %s" %model_filename)
     state_dict = torch.load(model_filename)
     model.load_state_dict(state_dict['model'])