from configs.config_base import config

config['gpu'] = '4'
config['lr'] = 0.0005
config['name'] = 'static_1_5'
config['notes'] = 'convnext + convnext, add dense with tanh at the end of reverse convnext, | \
    flattened latent dim to 128 | \
    train on 100 garments | \
    deeper model | \
    larger batch size | \
    ' 

config['cond_ae']['upscale_output'] = False
config['batch_size'] = 32


config['signature_dynamic'] = { 'id': (), }
config['signature_static'] = { 'uv_static': (512, 256, 3),
                         'mesh_static': (14475, 3,),
                         'mesh_mask': (14475,),
                         'uv_mask': (512, 256,),
                        }