Global settings typechecking (#148)

This commit is contained in:
Glenn Jocher
2023-01-05 15:13:22 +01:00
committed by GitHub
parent 19334ebb16
commit 3cbf3ec455
3 changed files with 26 additions and 5 deletions

View File

@ -48,7 +48,6 @@ class YOLO:
self.ckpt_path = None
self.cfg = None # if loaded from *.yaml
self.overrides = {} # overrides for trainer object
self.init_disabled = False # disable model initialization
# Load or create new YOLO model
{'.pt': self._load, '.yaml': self._new}[Path(model).suffix](model)