xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
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