Update ci.yaml (#37)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2022-11-07 19:27:55 +01:00
parent a733e41bcc
commit 523eff99e2
2 changed files with 41 additions and 6 deletions
--- a/ultralytics/yolo/v8/classify/train.py
+++ b/ultralytics/yolo/v8/classify/train.py
@ -58,7 +58,7 @@ def train(cfg):
 if __name__ == "__main__":
    """
    CLI usage:
-    python path/to/train.py epochs=10 project=PROJECT lr0=0.1
+    python ultralytics/yolo/v8/classify/train.py model=resnet18 data=imagenette160 epochs=1 img_size=224

    TODO:
    Direct cli support, i.e, yolov8 classify_train args.epochs 10