This commit is contained in:
2023-03-04 17:19:30 +00:00
parent 54e7d33065
commit b95ccc3ca1
5 changed files with 28 additions and 24 deletions
+5 -5
View File
@@ -1,7 +1,7 @@
config = dict()
config['gpu'] = '3'
config['name'] = 'iccv_pyr_32_64_128_256'
config['gpu'] = '0'
config['name'] = 'iccv_pyr_32_64_128_256_2'
config['notes'] = ''
config['tags'] = ['pyr', 'iccv']
@@ -22,13 +22,13 @@ 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'
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_2'
config['pyramid']['pretrained_pyramid_iteration'] = '125000'
config['pyramid']['pretrained_pyramid'] = ''
config['pyramid']['pretrained_pyramid_iteration'] = ''
config['pyramid']['final_upscale_enhancer'] = ''
config['train_dir'] = '/home/workspace/local-data/train'