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:
parent
c3d961fb03
commit
8f3cd52844
8 changed files with 77 additions and 12 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue