update baseline

2023-07-02 23:22:34 +02:00
parent 44103c6caa
commit 2fc0177d5a
3 changed files with 201 additions and 8 deletions
@@ -3,15 +3,15 @@ from utils.convnext_utils import CNBlockConfig
 config = dict()

 config['gpu'] = '2'
-config['name'] = 'abl_iccvw_all_1'
-config['notes'] = ''
+config['name'] = 'abl_iccvw_all_2'
+config['notes'] = 'fixed collision loss'
 config['tags'] = ['iccv', 'abl_all', 'torch']
 config['script'] = 'scripts/cvae.py'

-config['pretrained_model'] = ''
+config['pretrained_model'] = 'checkpoints/iccvw/abl_iccvw_all_1'

 config['use_skinning'] =              True # and deshaping
-config['use_normals'] =             True
+config['use_normals'] =               True
 config['use_extra_losses'] =          True # edge, collision, 3d mesh

 config['use_norm_loss'] =             True