making ready for skirts

configs/config_cvae.py

@@ -1,7 +1,7 @@
 from configs.config_base import config
 
-config['gpu'] = '4'
-config['name'] = 'cvae_low_4_3_d32'
+config['gpu'] = '2'
+config['name'] = 'cvae_19_skirt'
 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 \
@@ -11,13 +11,13 @@ remove shape from unposed garment\
 training on 100 garments with offsets over the average of these\
 fixed autoencoders!!!!!!!!!\
 cvae only loads encoders instead AEs\
-full res loss\
 '
-config['dataset_in_memory'] = False
+config['dataset_in_memory'] = True
 
 
-config['resolution'] = 32
-
+config['resolution'] = 512
+config['is_skirt'] = True
+config['reg_loss_coeff'] = 0 # set to 0 if don't want to use edge+collision
 
 if config['resolution'] != 512:
   config['notes'] += '\
@@ -50,13 +50,14 @@ if config['resolution'] != 512:
   config['signature_static']['uv_mask' + res_modifier] = (config['resolution'], config['resolution']//2)
 
 if config['resolution'] == 512:
-  config['avg_garment_path'] = 'custom_data/train_100_no_shape_avg_uv.npy'
+  config['avg_garment_path'] = 'custom_data/train_skirt_25_2_no_shape_avg_uv.npy'
 else:
   config['avg_garment_path'] = ''
 
-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['static_model'] = 'checkpoints/ae_static_7_2_skirt_x'
+config['body_model'] = 'checkpoints/ae_body_10_3_skirt_x'
+config['curvature_model'] = 'checkpoints/ae_curv_9_skirt_x'
+config['upscale_enhancer'] = 'checkpoints/up_2_3_skirt_x_d32'
 config['freeze_encoders'] = True
 
 config['cvae'] = dict()