{
    "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
        ]
    }
}