config.yaml

wandb:
    project_name: "test"
    project_team: "tail-upenn"
    online: false

project:
    device: 'gpu'
    num_workers: 12
    num_epochs: 1000
    pin_memory: True
    load_model: False
    val_interval: 10
    time: 0
    epoch: 0 

hyperparameters:
    batch_size: 8
    weights: True
    
    optimizer_fn: adam
        
    adam_learning_rate: .0003

    sgd_learning_rate: .01
    sgd_momentum: 0.9
    sgd_weight_decay: .0001
    sgd_nesterov: True

    loss_fn: focal 
    multiplier: [1, 1, 1, 1]
            
    tversky_alpha: .5
    tversky_beta: .5

    focal_alpha: 1
    focal_gamma: 2
    
    scheduler_fn: warm
    scheduler_patience: 3
    
    earlystopping: True
    earlystop_patience: 50
    earlystop_wait: 50

model:
    max_layer_size: 1024
    min_layer_size: 64

calibration:
    custom: False

image:
    phantom_format: 'dcm'
    mask_format: 'png'
    image_height: 600
    image_width: 360
    image_channels: 1
    mask_channels: 1
    mask_labels: 4

    labels: 
        0: 'background' 
        1: 'low_risk'
        2: 'moderate_risk'
        3: 'high_risk'

load:
    path: my_checkpoint.pth.tar

paths:
    train_img_dir: "data/train/phantom/"
    train_mask_dir: "data/train/mask/"
    val_img_dir: "data/val/phantom/"
    val_mask_dir: "data/val/mask/"
    test_img_dir: "data/test/phantom/"
    test_mask_dir: "data/test/mask/"
    predictions_dir: "data/predictions/"
    submissions_dir: "data/submissions/"
    calibrations_dir: "data/calibrations/"
    confidences_dir: "data/confidences/"