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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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)