This commit is contained in:
2023-03-04 18:35:35 +00:00
parent b95ccc3ca1
commit 76cb457a54
7 changed files with 580 additions and 11 deletions
@@ -1,7 +1,7 @@
config = dict()
config['gpu'] = '5'
config['name'] = 'iccv_pyr_32_64_128_skin'
config['name'] = 'iccv_pyr_32_64_128_skin_2'
config['notes'] = ''
config['tags'] = ['pyr', 'iccv']
@@ -22,17 +22,17 @@ 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_skin'
config['pretrained_model_iteration'] = ''
config['pyramid'] = dict()
config['pyramid']['resolutions'] = [32, 64, 128]
config['pyramid']['pretrained_pyramid'] = 'checkpoints/iccv/iccv_pyr_32_64_skin'
config['pyramid']['pretrained_pyramid_iteration'] = '130000'
config['pyramid']['pretrained_pyramid'] = ''
config['pyramid']['pretrained_pyramid_iteration'] = ''
config['pyramid']['final_upscale_enhancer'] = ''
config['train_dir'] = '/home/workspace/local-data_127/train'
config['valid_dir'] = '/home/workspace/local-data_127/valid'
config['train_dir'] = '/home/workspace/local-data/train'
config['valid_dir'] = '/home/workspace/local-data/valid'
config['test_dir'] = '/home/workspace/local-data_127/test_ready'
config['batch_size'] = 4
config['is_skirt'] = False