data_params:
  data_path: "dataset" 
  image_size: 224      

model_params:
  name: "efficientnet_b0"
  pretrained: True

train_params:
  epochs: 25
  batch_size: 64 
  optimizer: "AdamW"
  learning_rate: 0.001
  unfreeze_epoch: 5 

ddp_params:
  master_port: '12355'

output_params:
  save_dir: "runs/staging_classifier"
  checkpoint_name: "best_model.pth"