updated static training

This commit is contained in:
2023-07-01 10:46:43 +02:00
parent f9ac7fe3c3
commit cac04a1610
4 changed files with 13 additions and 11 deletions
+3 -3
View File
@@ -3,12 +3,12 @@ from utils.convnext_utils import CNBlockConfig
config = dict()
config['gpu'] = '3'
config['name'] = 'static_iccvw_2'
config['name'] = 'static_iccvw_4'
config['notes'] = ' '
config['tags'] = ['torch', 'static']
config['script'] = 'scripts/ae_static_train.py'
config['pretrained_model'] = 'checkpoints/iccvw/static_iccvw_1'
config['pretrained_model'] = 'checkpoints/iccvw/static_iccvw_3'
config['batch_size'] = 4
config['train_dir'] = '/data-local/127/cloth3d_processed'
@@ -17,7 +17,7 @@ config['test_dir'] = '/data-net/dataset/test_ready'
config['resolution'] = 512
config['is_skirt'] = False
config['dataset_in_memory'] = True
config['dataset_in_memory'] = False
config['is_cape'] = False
config['remove_global_orientation'] = True
config['device'] = 'cuda'