# @package _global_

# to execute this experiment run:
# python train.py experiment=catdog_ex

defaults:
  - override /paths: catdog
  - override /data: catdog
  - override /model: catdog_classifier
  - override /callbacks: default
  - override /logger: default
  - override /trainer: default

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

seed: 42
name: "catdog_experiment"

data:
  data_dir: "cats_and_dogs_filtered"
  batch_size: 64
  num_workers: 8
  pin_memory: True
  image_size: 224

model:
  lr: 5e-5
  weight_decay: 1e-5
  factor: 0.5
  patience: 5
  min_lr: 1e-6
  num_classes: 2
  patch_size: 16
  embed_dim: 256
  depth: 4
  num_heads: 4
  mlp_ratio: 4

trainer:
  min_epochs: 1
  max_epochs: 5

callbacks:
  model_checkpoint:
    monitor: "val_acc"
    mode: "max"
    save_top_k: 1
    save_last: True
    verbose: True

  early_stopping:
    monitor: "val_acc"
    patience: 10
    mode: "max"
    verbose: True

  rich_model_summary:
    max_depth: 1

  rich_progress_bar:
    refresh_rate: 1