from utils.convnext_utils import CNBlockConfig

config = dict()

config['batch_size'] = 4
config['is_cape'] = False
config['device'] = 'cuda'

config['train_dir'] = '/data-local/127/cloth3d_processed'
config['valid_dir'] = '/data-local/127/cloth3d_processed'
config['test_dir'] = '/data-local/127_2/cloth3d_processed_test'

config['signature']: set = { 'id',
                            'frame_nr',
                            'pose_shape',
                            'uv_body_posed',
                            'mesh_body_posed',
                            'mesh_posed',
                            'mesh_unposed',
                            'uv_unposed',
                            'uv_posed',
                            'uv_normals',
                            'uv_static',
                            'mesh_mask',
                            'uv_mask',
                        }



for res in [64, 128, 256]:
    res_modifier = '_' + str(res)
    config['signature'].add('uv_body_posed' + res_modifier)
    config['signature'].add('uv_unposed' + res_modifier)
    config['signature'].add('uv_posed' + res_modifier)
    config['signature'].add('uv_normals' + res_modifier)
    config['signature'].add('uv_static' + res_modifier)
    config['signature'].add('uv_mask' + res_modifier)