weights/CAMUS/dataset_config.json

{
    "DATASET": "../Dataset/CAMUS/Landmarks_3_10",
    "inputsize": 512,
    "name": "CAMUS_NN_dual_no_raster",
    "dataset": "../Dataset/CAMUS/Landmarks_3_10",
    "resume": "../Training/CAMUS_Landmarks_3_10_PreTrain_Dual/CAMUS_Landmarks_3_10_PreTrain_Dual.pth",
    "iterations": 500000,
    "epochs": 32,
    "batch_size": 1,
    "val_batch_size": 1,
    "warm_up_it": 100000,
    "lr": 0.0001,
    "weight_decay": 1e-06,
    "gamma": 0.1,
    "chamfer_w": 10.0,
    "kld_w": 0.001,
    "dice_w": 1.0,
    "bce_w": 1.0,
    "conv_dice_w": 1.0,
    "conv_bce_w": 1.0,
    "edge_w": 1.0,
    "elasticity_w": 250.0,
    "curvature_w": 300.0,
    "latents": 64,
    "initial_filters": 16,
    "flip_h": false,
    "flip_v": false,
    "transpose": false,
    "rotate": true,
    "raster_as_input": false,
    "use_dual": true,
    "use_raster": false,
    "naive": false,
    "val_every": 1,
    "production": false,
    "stepsize": 16
}