Fix yolo mode=train CLI bug on model load (#133)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher
2023-01-02 19:24:44 +01:00
committed by GitHub
parent c3d961fb03
commit 8f3cd52844
8 changed files with 77 additions and 12 deletions

View File

@ -362,7 +362,7 @@ class BaseTrainer:
return
# We should improve the code flow here. This function looks hacky
model = self.model
pretrained = not (str(model).endswith(".yaml"))
pretrained = not str(model).endswith(".yaml")
# config
if not pretrained:
model = check_file(model)