week1_image_classifier/config.yaml

# Dataset Configuration
dataset:
  name: "cifar10"  # Changed from food101
  train_split: "train"
  test_split: "test"
  val_split_ratio: 0.1
  
# Model Configuration
model:
  name: "microsoft/resnet-50"
  num_labels: 10  # CIFAR-10 has 10 classes