This commit is contained in:
2023-03-04 20:32:16 +00:00
parent 76cb457a54
commit dfdf8d408e
4 changed files with 17 additions and 16 deletions
+7 -7
View File
@@ -1,7 +1,7 @@
config = dict()
config['gpu'] = '4'
config['name'] = 'iccv_pyr_32_64_128_reg'
config['gpu'] = '1'
config['name'] = 'iccv_pyr_32_64_128_reg_2'
config['notes'] = ''
config['tags'] = ['pyr', 'iccv', 'reg']
@@ -22,17 +22,17 @@ config['use_offsets'] = False
config['use_norm_loss'] = False
config['use_curvature'] = True
config['pretrained_model'] = ''
config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_32_64_128_reg'
config['pretrained_model_iteration'] = ''
config['pyramid'] = dict()
config['pyramid']['resolutions'] = [32, 64, 128]
config['pyramid']['pretrained_pyramid'] = 'checkpoints/iccv/iccv_pyr_32_64_reg'
config['pyramid']['pretrained_pyramid_iteration'] = '120000'
config['pyramid']['pretrained_pyramid'] = ''
config['pyramid']['pretrained_pyramid_iteration'] = ''
config['pyramid']['final_upscale_enhancer'] = ''
config['train_dir'] = '/home/workspace/local-data/train'
config['valid_dir'] = '/home/workspace/local-data/valid'
config['train_dir'] = '/home/workspace/local-data_127/train'
config['valid_dir'] = '/home/workspace/local-data_127/valid'
config['test_dir'] = '/home/workspace/local-data_127/test_ready'
config['batch_size'] = 4
config['is_skirt'] = False