Add colab template and minor fixes (#137)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-04 02:00:39 +05:30
parent 3c18aa5278
commit dcd8ef68e6
4 changed files with 18 additions and 6 deletions
--- a/ultralytics/yolo/engine/model.py
+++ b/ultralytics/yolo/engine/model.py
@ -224,4 +224,5 @@ class YOLO:
        args.pop("device", None)
        args.pop("project", None)
        args.pop("name", None)
-        args.pop("batch_size", None)
+        args.pop("batch", None)
+        args.pop("epochs", None)