configs/resnet50.yaml

# Model configurations
model:
  name: "resnet50"
  pretrained: true
  in_chans: 1
  num_classes: 1

# Training configurations
training:
  batch_size: 16
  learning_rate: 0.00005
  optimizer: "adam"
  max_epochs: 50

# Dataset configurations
dataset:
  root_dir: "/home/venom/Documents/x_ray/"
  label_csv: "data/labels.csv"
  val_split: 0.05
  random_seed: 35
  apply_equalization: false
  batch_size: 16

# Logging configurations
logging:
  log_dir: "./logs/"
  log_every_n_steps: 1