update main script

configs/config_cvae.py

@@ -1,7 +1,7 @@
 from configs.config_base import config
 
-config['gpu'] = '4'
-config['name'] = 'v2_4'
+config['gpu'] = '0'
+config['name'] = 'v2_7_4_3'
 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 | \
@@ -12,6 +12,13 @@ training on 100 garments with offsets over the average of these | \
 fixed autoencoders!!!!!!!!! | \
 cvae only loads encoders instead AEs | \
 increase balancing coeff sooner | \
+use 3d mesh loss | \
+add final gc layer | \
+starting from v2_5_2, add tanh to cond_enc last layer | \
+NOT using one-hot skinning | \
+using new average garment with one-hot versions | \
+fix collision for plotting | \
+use 3d mesh loss | \
 '
 config['dataset_in_memory'] = False
 
@@ -21,18 +28,23 @@ config['dataset_in_memory'] = False
 # 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
+# DON'T FORGET TO CHECK AVG GARMENT USED
 
+config['pretrained_model'] = 'checkpoints/v2_5_2'
 
 config['resolution'] = 512
 config['is_skirt'] = False
 config['reg_loss_coeff'] = 0.1 # set to 0 if don't want to use edge+collision
+config['l2_coeff'] = 1
 
 if config['resolution'] != 512:
   config['notes'] += '\
 using custom enhancer for upscaling, using not filled versions\
 '
     
-config['lr'] = 0.0001
+###############    
+config['lr'] = 0.0001#                 0.0001
+###############
 config['starting_balancing_coeff'] = 0.0001
 
 config['signature_dynamic'] = { 'id': (),
@@ -58,7 +70,7 @@ 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_no_shape_avg_uv.npy'
+  config['avg_garment_path'] = 'custom_data/train_no_shape_avg_uv.npy' # train_no_shape_avg_uv  train_no_shape_one_hot_avg_uv
 else:
   config['avg_garment_path'] = ''
 
@@ -67,10 +79,13 @@ config['body_model'] = 'checkpoints/body_1'
 config['curvature_model'] = 'checkpoints/curv_1'
 config['upscale_enhancer'] = ''
 config['freeze_encoders'] = True
+config['use_one_hot'] = False
 
 config['cvae'] = dict()
 config['cvae']['is_label'] = False
 config['cvae']['latent_dim'] = config['latent_dim']
+config['cvae']['add_noise'] = True
+config['cvae']['noise_amount'] = 0.025
 
 if config['resolution'] == 512:
   config['cvae']['kernel_size'] = 7