training_config.yaml

# Dataset split
train_split: 0.9
val_split: 0.1

# Training parameters
train_params:
  epochs: 50
  batch: 16
  imgsz: 640
  lr0: 0.01
  lrf: 0.01