save configs

configs/config_cvae.py

@@ -1,23 +1,31 @@
 from configs.config_base import config
 
-config['gpu'] = '2'
-config['name'] = 'cvae_19_skirt'
+config['gpu'] = '4'
+config['name'] = 'v2_4'
 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\
+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 | \
+increase balancing coeff sooner | \
 '
-config['dataset_in_memory'] = True
+config['dataset_in_memory'] = False
+
+
+# DON'T FORGET TO RESET SSIM LOSS USAGE
+# DON'T FORGET TO CHECK MODEL LOADING
+# DON'T FORGET TO CHECK IF STORE IN MEMORY
+# DON'T FORGET TO CHECK REG LOSS COEFF
+# DON'T FORGET TO LOAD FINAL UPSCALER UNLESS HAVE LEVEL 512
 
 
 config['resolution'] = 512
-config['is_skirt'] = True
-config['reg_loss_coeff'] = 0 # set to 0 if don't want to use edge+collision
+config['is_skirt'] = False
+config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
 
 if config['resolution'] != 512:
   config['notes'] += '\
@@ -50,14 +58,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_skirt_25_2_no_shape_avg_uv.npy'
+  config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy'
 else:
   config['avg_garment_path'] = ''
 
-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['static_model'] = 'checkpoints/static_1'
+config['body_model'] = 'checkpoints/body_1'
+config['curvature_model'] = 'checkpoints/curv_1'
+config['upscale_enhancer'] = ''
 config['freeze_encoders'] = True
 
 config['cvae'] = dict()