bhimrazy's picture
updates config and removes old logs
2c1d9fe
raw
history blame
453 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: false
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