config.yaml

device: "auto" # "cuda", "cpu", or "auto"
classes: ["glass", "paper", "cardboard", "plastic", "metal", "trash"]
split: [0.70, 0.15, 0.15]
img_size: [224, 224]
batch_size: 32
epochs: 25
patience: 5
lr: 0.001
random_seed: 42