save

evaluate_ae.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.config_abl_all_post import config
+from configs.iccv.config_abl_all import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -103,11 +103,13 @@ if __name__ == '__main__':
     ids = set(ids)
 
     iteration = ''
-
-    load_model(model, 'checkpoints/post/' + checkpoint_name, iteration=iteration)
+    if iteration == '':
+        iteration = load_model(model, 'checkpoints/iccv/' + checkpoint_name, return_iteration=True)
+    else:
+        load_model(model, 'checkpoints/iccv/' + checkpoint_name, iteration=iteration)
     # load_model(model, 'checkpoints/' + 'abl_all_120k')
 
-    output_folder = '/home/workspace/data/results/post/' + checkpoint_name + '-' + iteration
+    output_folder = '/home/workspace/local-data_128/results/iccv/' + checkpoint_name + '-' + iteration
 
 
     os.makedirs(output_folder, exist_ok=True)