defaults:
  - agent: resnet_diffusion
  - data: resnet_dataset
  - eval: eval
  - _self_

train:
  batch_size: 16       # Batch size for training
  lr: 1e-4            # Learning rate
  max_steps: 20000    # Maximum training steps
  log_freq: 100       # Log frequency (steps)
  save_freq: 2000     # Save checkpoint frequency (steps)
  device: "cuda"      # Device: "cuda" or "cpu"
  num_workers: 8      # DataLoader workers (set to 0 for debugging, 8 for production)