HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.0.29 DDP-cls and default arg fixes (#813)

Browse source
This commit is contained in:
Glenn Jocher 2023-02-06 02:30:03 +04:00 • committed by GitHub
parent 21ae321bc2
commit 7a7c8dc7b7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 38 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

7
ultralytics/yolo/engine/model.py
View file

@ -119,7 +119,6 @@ class YOLO:
def fuse(self):
self.model.fuse()
@smart_inference_mode()
def predict(self, source=None, stream=False, **kwargs):
"""
Perform prediction using the YOLO model.
@ -258,8 +257,6 @@ class YOLO:
@staticmethod
def _reset_ckpt_args(args):
for arg in 'verbose', 'project', 'name', 'exist_ok', 'resume', 'batch', 'epochs', 'cache', 'save_json', \
'half', 'v5loader':
for arg in 'augment', 'verbose', 'project', 'name', 'exist_ok', 'resume', 'batch', 'epochs', 'cache', \
'save_json', 'half', 'v5loader', 'device', 'cfg', 'save', 'rect', 'plots':
args.pop(arg, None)
args["device"] = '' # set device to '' to prevent auto-DDP usage