only load encoders of AEs
This commit is contained in:
+17
-4
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user