This commit is contained in:
2023-01-29 00:37:44 +00:00
parent 0dfe7f7659
commit 1f8793e4e8
2 changed files with 3 additions and 2 deletions
+1
View File
@@ -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
+2 -2
View File
@@ -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')