bhimrazy's picture
Updates the model and config
09ebe4d
raw
history blame contribute delete
No virus
452 Bytes
# 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