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

2
tests/test_cuda.py
View file

@ -71,7 +71,7 @@ def test_predict_sam():
predictor = SAMPredictor(overrides=overrides)
# Set image
predictor.set_image('ultralytics/assets/zidane.jpg') # set with image file
predictor.set_image(ASSETS / 'zidane.jpg') # set with image file
# predictor(bboxes=[439, 437, 524, 709])
# predictor(points=[900, 370], labels=[1])