luni/cvpr_2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update training scripts

Browse Source
This commit is contained in:
luni
2023-06-30 18:29:03 +02:00
parent 8639992cd8
commit c29654735c
11 changed files with 552 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/ae_body_cape_m.py
+4 -4
View File
@@ -2,7 +2,7 @@ from utils.convnext_utils import CNBlockConfig
config = dict()
config['gpu'] = '4'
config['gpu'] = '0'
config['name'] = 'body_torch_cape_m'
config['notes'] = ' '
config['tags'] = ['torch', 'body']
@@ -11,9 +11,9 @@ config['script'] = 'scripts/ae_body_train.py'
config['pretrained_model'] = ''
config['batch_size'] = 4
config['train_dir'] = '/data-local/127/cape_processed/train/male'
config['valid_dir'] = '/data-local/127/cape_processed/test/male'
config['test_dir'] = '/data-local/127/cape_processed/test/male'
config['train_dir'] = '/data-local/128/cape_processed/train/male'
config['valid_dir'] = '/data-local/128/cape_processed/test/male'
config['test_dir'] = '/data-local/128/cape_processed/test/male'
config['train_frames'] = None
config['valid_frames'] = None