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

Failing CUDA tests fixes (#4682)

Browse source
This commit is contained in:
Glenn Jocher 2023-09-01 13:29:08 +02:00 • committed by GitHub
parent 263bfd1e93
commit 2bc6e647c7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_python.py
View file

@ -140,11 +140,11 @@ def test_track_stream():
# Test Global Motion Compensation (GMC) methods
for gmc in 'orb', 'sift', 'ecc':
with open(ROOT / 'cfg/trackers/botsort.yaml') as f:
with open(ROOT / 'cfg/trackers/botsort.yaml', encoding='utf-8') as f:
data = yaml.safe_load(f)
tracker = TMP / f'botsort-{gmc}.yaml'
data['gmc_method'] = gmc
with open(tracker, 'w') as f:
with open(tracker, 'w', encoding='utf-8') as f:
yaml.safe_dump(data, f)
model.track('https://ultralytics.com/assets/decelera_portrait_min.mov', imgsz=160, tracker=tracker)
@ -166,7 +166,7 @@ def test_train_pretrained():
def test_export_torchscript():
f = YOLO(MODEL).export(format='torchscript', optimize=True)
f = YOLO(MODEL).export(format='torchscript', optimize=False)
YOLO(f)(SOURCE) # exported model inference