File size: 452 Bytes
a2f341c 2c1d9fe 09ebe4d a2f341c 2c1d9fe a2f341c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# data:
train_csv_path: data/diabetic-retinopathy-dataset/train.csv
val_csv_path: data/diabetic-retinopathy-dataset/val.csv
# experiment:
seed: 42
batch_size: 128
num_workers: 32
use_class_weighting: true
use_weighted_sampler: false # class weighting and weighted sampler are mutually exclusive
model_name: "densenet169"
max_epochs: 20
image_size: 224
learning_rate: 3e-4
use_scheduler: true
logs_dir: logs
checkpoint_dirpath: artifacts/checkpoints
|