lot more training

configs/iccv/config_abl_pyr_32_skirt.py

@@ -1,6 +1,6 @@
 config = dict()
 
-config['gpu'] = '1'
+config['gpu'] = '4'
 config['name'] = 'iccv_pyr_32_skirt_3'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv']
@@ -20,6 +20,7 @@ config['remove_global_orientation'] = True
 config['use_extra_losses'] =          True # edge, collision, 3d mesh
 config['use_offsets'] =               False
 config['use_norm_loss'] =             True
+config['use_curvature'] =             True
 
 config['pretrained_model'] = ''
 config['pretrained_model_iteration'] = ''
@@ -31,7 +32,7 @@ config['pyramid']['final_upscale_enhancer'] = ''
 
 config['train_dir'] = '/home/workspace/local-data_128/train_skirt'
 config['valid_dir'] = '/home/workspace/local-data_127/valid_skirt'
-config['test_dir'] = '/home/workspace/data/dataset/test_skirt_ready_2'
+config['test_dir'] = '/home/workspace/local-data/test_skirt_ready_2'
 config['batch_size'] = 4
 config['is_skirt'] = True
 config['is_cape'] = False