From 1f8793e4e8328b85f19f049d1757fd1ee2472776 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sun, 29 Jan 2023 00:37:44 +0000 Subject: [PATCH] evaluate --- aaa_helpers.txt | 1 + evaluate_ae_pyramid.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) 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')