22 lines
679 B
Python
22 lines
679 B
Python
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,),
|
|
} |