only load encoders of AEs

This commit is contained in:
2022-09-26 11:35:08 +00:00
parent 55f30c0fa9
commit fdf36b27e7
3 changed files with 32 additions and 44 deletions
+17 -4
View File
@@ -1,7 +1,20 @@
from configs.config_base import config
config['lr'] = 0.0001
config['gpu'] = '3'
config['name'] = 'cvae_19'
config['notes'] = '\
offsets over average unposed garment. remove global orientation (in posed body input and in pose_shape parameters. \
retrained body encoder with deoriented bodies \
use mean l2 loss. small batch. l2 * 1000, balancing to 1 from 0.01\
fixed gender issue and retrained body encoder, FINAL\
remove shape from unposed garment\
training on 100 garments with offsets over the average of these\
fixed autoencoders!!!!!!!!!\
cvae only loads encoders instead AEs\
'
config['lr'] = 0.0001
config['starting_balancing_coeff'] = 0.0001
config['signature_dynamic'] = { 'id': (),
@@ -19,9 +32,9 @@ config['signature_static'] = { 'uv_static': (512, 256, 3),
}
config['avg_garment_path'] = 'custom_data/train_100_no_shape_avg_uv.npy'
config['static_model'] = 'checkpoints/ae_static_6_copy'
config['body_model'] = 'checkpoints/ae_body_8_copy'
config['curvature_model'] = 'checkpoints/ae_curvature_7_copy'
config['static_model'] = 'checkpoints/ae_static_7_2_x'
config['body_model'] = 'checkpoints/ae_body_9_x'
config['curvature_model'] = 'checkpoints/ae_curv_9_x'
config['freeze_encoders'] = True
config['cvae'] = dict()