diff --git a/aaa_helpers.txt b/aaa_helpers.txt index aebd959..92e8934 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -1,4 +1,5 @@ nohup python cvae_pyramid.py --name="lhunor_pyr_abl_32_64_e2e_r3" > nohup/review/pyr_abl_32_64_e2e_r3.out 2>&1 & +nohup python evaluate_ae_pyramid.py --name evaluate_pyr_abl_32_64_e2e_r3 > nohup/evaluate/pyr_abl_32_64_e2e_r3.out 2>&1 & one final convolutional layer for smoothing, on the final coordinates diff --git a/evaluate_ae_pyramid.py b/evaluate_ae_pyramid.py index 730e000..283477c 100644 --- a/evaluate_ae_pyramid.py +++ b/evaluate_ae_pyramid.py @@ -1,6 +1,6 @@ import os import sys -from configs.config_abl_2_pyr_32_64_128_256_512 import config +from configs.config_abl_pyr_32 import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] import tensorflow as tf @@ -97,7 +97,7 @@ if __name__ == '__main__': ids = [id.split('_')[0] for id in ids] ids = set(ids) - iteration = '50000' + iteration = '200000' load_model(model, 'checkpoints/' + checkpoint_name, iteration=iteration) # load_model(model, 'checkpoints/' + 'abl_all_120k')