add no curvature option

This commit is contained in:
2023-01-26 12:03:50 +00:00
parent 2ac1a812fc
commit 44e1d02e5a
4 changed files with 32 additions and 19 deletions
+6 -3
View File
@@ -1,8 +1,8 @@
config = dict()
config['gpu'] = '3'
config['name'] = 'pyr_abl_32_skirt_r1'
config['notes'] = ''
config['gpu'] = '6'
config['name'] = 'pyr_abl_32_skirt_no_curv_r1'
config['notes'] = 'review_1'
# DON'T FORGET TO RESET SSIM LOSS USAGE
# DON'T FORGET TO CHECK MODEL LOADING
@@ -18,6 +18,7 @@ config['use_skinning'] = True # and deshaping
config['remove_global_orientation'] = True
config['use_extra_losses'] = True # edge, collision, 3d mesh
config['use_offsets'] = False
config['use_curvature'] = False
config['pyramid'] = dict()
config['pyramid']['resolutions'] = [32]
@@ -33,6 +34,8 @@ config['test_dir'] = '/home/workspace/data/dataset/test_skirt_ready_2'
config['batch_size'] = 4
config['is_skirt'] = True
config['is_cape'] = False
config['frames_file'] = 'frames_skirt_10p.txt'
config['frames_file_valid'] = 'frames_skirt_10p.txt'
config['pyramid']['static_model'] = 'checkpoints/static_1_skirt'
config['pyramid']['body_model'] = 'checkpoints/body_1_skirt_2_2'