CV-Lab/hyper.yaml at main · liu-bodong/CV-Lab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
# hyper.yaml
# This is the configuration file for training

# wandb configuration
wandb_project: unet
wandb_entity: liubodong-cornell-university  # Replace with your actual entity name
wandb_mode: online  # Set to 'offline' if you want to run without logging to Weights & Biases

# Model configuration
model_type: UNet
image_size: [256, 256]
input_channels: 3
output_channels: 1
channels: [64, 128, 256, 512]

# Training configuration
batch_size: 8
epochs: 100
init_lr: 0.0003
lr_rampdown_epochs: 120
val_split: 0.2
optimizer: Adam
loss: BCEWithLogitsLoss
save_after_epochs: 50
patience: 20
min_improvement: 0.0001
use_amp: true

# Other
device: cuda
gpu_monitoring: true
data_dir: ./data/brain_MRI_Buda/kaggle_3m
dataset: BrainMRIDataset
save_dir: ./saved_models