save config files

configs/config_cvae_2.json

@@ -0,0 +1 @@
+{"batch_size": 4, "train_dir": "/home/workspace/local-data/train", "valid_dir": "/home/workspace/local-data/valid", "test_dir": "/home/workspace/data/dataset/test_ready", "use_filled": false, "resolution": 512, "is_skirt": false, "dataset_in_memory": false, "is_cape": false, "smpl_res": 14475, "latent_dim": 128, "cond_ae": {"is_label": false, "upscale_output": false, "latent_dim": 128, "kernel_size": 7, "bottleneck_size": 64, "encoder_depths": [3, 3, 9, 9, 3], "encoder_dims": [32, 64, 128, 256, 64], "decoder_input": [8, 4, 64], "decoder_depths": [3, 3, 9, 3, 3], "decoder_dims": [64, 256, 128, 64, 16]}, "cond_ae_body": {"latent_dim": 128, "kernel_size": 7, "upscale_output": true, "bottleneck_size": 512, "encoder_depths": [3, 3, 9, 9, 3, 3], "encoder_dims": [32, 64, 128, 256, 256, 512], "decoder_input": [4, 2, 512], "decoder_depths": [3, 3, 9, 3], "decoder_dims": [512, 128, 64, 16]}, "gpu": "0", "name": "v2_7_4_3", "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 | 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 | ", "pretrained_model": "checkpoints/v2_5_2", "reg_loss_coeff": 0.1, "l2_coeff": 1, "lr": 0.0001, "starting_balancing_coeff": 0.0001, "signature_dynamic": {"id": [], "pose_shape": [83], "uv_body_posed": [512, 256, 3], "mesh_body_posed": [14475, 3], "mesh_posed": [14475, 3], "mesh_unposed": [14475, 3], "uv_unposed": [512, 256, 3], "uv_curvature": [512, 256, 7]}, "signature_static": {"uv_static": [512, 256, 3], "mesh_mask": [14475], "uv_mask": [512, 256]}, "avg_garment_path": "custom_data/train_no_shape_avg_uv.npy", "static_model": "checkpoints/static_1", "body_model": "checkpoints/body_1", "curvature_model": "checkpoints/curv_1", "upscale_enhancer": "", "freeze_encoders": true, "use_one_hot": false, "cvae": {"is_label": false, "latent_dim": 128, "add_noise": true, "noise_amount": 0.025, "kernel_size": 7, "encoder_depths": [3, 3, 9, 3], "encoder_dims": [32, 64, 128, 256], "decoder_input": [16, 8, 80], "decoder_depths": [3, 3, 9, 3], "decoder_dims": [256, 128, 64, 16]}}