190 lines
4.3 KiB
JSON
190 lines
4.3 KiB
JSON
{
|
|
"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
|
|
]
|
|
}
|
|
} |