From 19e868f2716cfdd719ebb5fe9704f3ba25ef21e4 Mon Sep 17 00:00:00 2001 From: HunorLaczko Date: Tue, 23 May 2023 10:35:14 +0000 Subject: [PATCH] save --- aaa_helpers.txt | 2 +- configs/iccv/config_abl_pyr_32_64_128_256_seq.py | 8 ++++---- cvae_pyramid.py | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 7708ad7..c517b66 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -127,7 +127,7 @@ nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_skin" > nohup nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_curv" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_curv.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_curv_skirt" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_curv_skirt.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_reg" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_reg.out 2>&1 & -nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_seq" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_seq.out 2>&1 & +nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_seq_2_fine" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_seq_2_fine.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_512" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_512.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_64_128_256_512_skirt" > nohup/iccv/pyr/iccv_pyr_32_64_128_256_512_skirt.out 2>&1 & diff --git a/configs/iccv/config_abl_pyr_32_64_128_256_seq.py b/configs/iccv/config_abl_pyr_32_64_128_256_seq.py index 4ddab6d..317f798 100644 --- a/configs/iccv/config_abl_pyr_32_64_128_256_seq.py +++ b/configs/iccv/config_abl_pyr_32_64_128_256_seq.py @@ -1,7 +1,7 @@ config = dict() -config['gpu'] = '3' -config['name'] = 'iccv_pyr_32_64_128_256_seq_2' +config['gpu'] = '1' +config['name'] = 'iccv_pyr_32_64_128_256_seq_2_fine' config['notes'] = '' config['tags'] = ['pyr', 'iccv'] @@ -22,12 +22,12 @@ config['use_offsets'] = False config['use_norm_loss'] = True config['use_curvature'] = True -config['pretrained_model'] = '' +config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_32_64_128_256_seq_2' config['pretrained_model_iteration'] = '' config['pyramid'] = dict() config['pyramid']['resolutions'] = [32, 64, 128, 256] -config['pyramid']['pretrained_pyramid'] = 'checkpoints/iccv/iccv_pyr_32_64_128_seq_2' +config['pyramid']['pretrained_pyramid'] = '' config['pyramid']['final_upscale_enhancer'] = '' config['train_dir'] = '/home/workspace/local-data/train' diff --git a/cvae_pyramid.py b/cvae_pyramid.py index 3615ab6..38b44a2 100644 --- a/cvae_pyramid.py +++ b/cvae_pyramid.py @@ -1,6 +1,6 @@ import os import sys -from configs.iccv.config_abl_pyr_32_tshirt import config +from configs.iccv.config_abl_pyr_32_64_128_256_seq import config os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']