weights/MRI/dataset_config.json

{
    "DATASET": "../Dataset/Sunnybrook/Landmarks_3_10",
    "inputsize": 256,
    "name": "Sunnybrook_dual_3",
    "dataset": "../Dataset/Sunnybrook/Landmarks_3_10",
    "iterations": 500000,
    "epochs": 770,
    "batch_size": 1,
    "val_batch_size": 1,
    "warm_up_it": 50000,
    "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": true,
    "flip_v": true,
    "transpose": true,
    "rotate": true,
    "raster_as_input": false,
    "use_dual": true,
    "use_raster": true,
    "naive": true,
    "val_every": 10,
    "pretraining": false,
    "production": false,
    "stepsize": 385
}