luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

save

Browse Source
This commit is contained in:
luni
2023-03-14 15:35:32 +00:00
parent d92a18e212
commit 30e33bb037
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/iccv/config_abl_pyr_32_64_128_seq.py
+2 -2
View File
@@ -1,6 +1,6 @@
config = dict() config = dict()
config['gpu'] = '4' config['gpu'] = '1'
config['name'] = 'iccv_pyr_32_64_128_seq_2' config['name'] = 'iccv_pyr_32_64_128_seq_2'
config['notes'] = '' config['notes'] = ''
config['tags'] = ['pyr', 'iccv'] config['tags'] = ['pyr', 'iccv']
@@ -32,7 +32,7 @@ config['pyramid']['final_upscale_enhancer'] = ''
config['train_dir'] = '/home/workspace/local-data/train' config['train_dir'] = '/home/workspace/local-data/train'
config['valid_dir'] = '/home/workspace/local-data/valid' config['valid_dir'] = '/home/workspace/local-data/valid'
config['test_dir'] = '/home/workspace/local-data_127/test_ready' config['test_dir'] = '/home/workspace/local-data/test_ready'
config['batch_size'] = 4 config['batch_size'] = 4
config['is_skirt'] = False config['is_skirt'] = False
configs/iccv/config_abl_pyr_32_seq.py
+1 -1
View File
@@ -32,7 +32,7 @@ config['pyramid']['final_upscale_enhancer'] = ''
config['train_dir'] = '/home/workspace/local-data/train' config['train_dir'] = '/home/workspace/local-data/train'
config['valid_dir'] = '/home/workspace/local-data/valid' config['valid_dir'] = '/home/workspace/local-data/valid'
config['test_dir'] = '/home/workspace/local-data_127/test_ready' config['test_dir'] = '/home/workspace/local-data/test_ready'
config['batch_size'] = 4 config['batch_size'] = 4
config['is_skirt'] = False config['is_skirt'] = False
evaluate_ae_pyramid.py
+1 -1
View File
@@ -1,6 +1,6 @@
import os import os
import sys import sys
from configs.iccv.config_abl_pyr_32_64_128_256_seq import config from configs.iccv.config_abl_pyr_32_64_128_seq import config
os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
import tensorflow as tf import tensorflow as tf