weights/PAXRAY/dataset_config.json

{
    "DATASET": "../Dataset/PAXRay_Front/Landmarks_3_10",
    "inputsize": 512,
    "name": "PRF_dual_no_raster_NN_v3",
    "dataset": "../Dataset/PAXRay_Front/Landmarks_3_10",
    "resume": "../Training/PRF_dual_no_raster_NN_v2/PRF_dual_no_raster_NN_v2.pth",
    "iterations": 600000,
    "epochs": 101,
    "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": 2.0,
    "elasticity_w": 1000.0,
    "curvature_w": 500.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": 10,
    "pretraining": true,
    "production": false,
    "stepsize": 50
}