train YOLOv8-S with 300 epoch

yolo/config/__init__.py

@@ -4,7 +4,6 @@ from .yolov2_config    import build_yolov2_config
 from .yolov3_config    import build_yolov3_config
 from .yolov5_config    import build_yolov5_config
 from .yolov5_af_config import build_yolov5af_config
-from .yolov7_af_config import build_yolov7af_config
 from .yolov8_config    import build_yolov8_config
 from .gelan_config     import build_gelan_config
 from .rtdetr_config    import build_rtdetr_config
@@ -24,8 +23,6 @@ def build_config(args):
         cfg = build_yolov5af_config(args)
     elif 'yolov5' in args.model:
         cfg = build_yolov5_config(args)
-    elif 'yolov7_af' in args.model:
-        cfg = build_yolov7af_config(args)
     elif 'yolov8' in args.model:
         cfg = build_yolov8_config(args)
     elif 'gelan' in args.model:

yolo/models/__init__.py

@@ -7,7 +7,6 @@ from .yolov2.build    import build_yolov2
 from .yolov3.build    import build_yolov3
 from .yolov5.build    import build_yolov5
 from .yolov5_af.build import build_yolov5af
-from .yolov7_af.build import build_yolov7af
 from .yolov8.build    import build_yolov8
 from .gelan.build     import build_gelan
 from .rtdetr.build    import build_rtdetr
@@ -31,9 +30,6 @@ def build_model(args, cfg, is_val=False):
     ## Modified YOLOv5
     elif 'yolov5' in args.model:
         model, criterion = build_yolov5(cfg, is_val)
-    ## Anchor-free YOLOv7
-    elif 'yolov7_af' in args.model:
-        model, criterion = build_yolov7af(cfg, is_val)
     ## YOLOv8
     elif 'yolov8' in args.model:
         model, criterion = build_yolov8(cfg, is_val)