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

TensorRT Export Fix (#171)

Browse source
This commit is contained in:
Glenn Jocher 2023-01-10 01:00:44 +01:00 • committed by GitHub
parent 422c49d439
commit 202f7bffa3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 12 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/configs/default.yaml
View file

@ -13,7 +13,7 @@ batch: 16 # number of images per batch
imgsz: 640 # size of input images
save: True # save checkpoints
cache: False # True/ram, disk or False. Use cache for data loading
device: '' # cuda device, i.e. 0 or 0,1,2,3 or cpu. Device to run on
device: null # cuda device, i.e. 0 or 0,1,2,3 or cpu. Device to run on
workers: 8 # number of worker threads for data loading
project: null # project name
name: null # experiment name