Hydra *.yml extension deprecated fix (#34)

ultralytics/yolo/engine/trainer.py

@@ -24,7 +24,7 @@ from ultralytics.yolo.utils import LOGGER, ROOT
 from ultralytics.yolo.utils.files import increment_path, save_yaml
 from ultralytics.yolo.utils.modeling import get_model
 
-DEFAULT_CONFIG = ROOT / "yolo/utils/configs/default.yml"
+DEFAULT_CONFIG = ROOT / "yolo/utils/configs/default.yaml"
 
 
 class BaseTrainer:

ultralytics/yolo/v8/classify/train.py

@@ -47,7 +47,7 @@ class ClassificationTrainer(BaseTrainer):
         return torch.nn.functional.cross_entropy(preds, targets)
 
 
-@hydra.main(version_base=None, config_path=DEFAULT_CONFIG.parent, config_name=DEFAULT_CONFIG.stem)
+@hydra.main(version_base=None, config_path=DEFAULT_CONFIG.parent, config_name=DEFAULT_CONFIG.name)
 def train(cfg):
     cfg.model = cfg.model or "resnet18"
     cfg.data = cfg.data or "imagenette160"  # or yolo.ClassificationDataset("mnist")