GPU export options (#2297)
* option for skip last layer and cuda export support
* added parameter device
* fix import
* cleanup 1
* cleanup 2
* opt-in grid
--grid will export with grid computation, default export will skip grid (same as current)
* default --device cpu
GPU export causes ONNX and CoreML errors.
Co-authored-by:
Jan Hajek <jan.hajek@gmail.com>
Co-authored-by:
Glenn Jocher <glenn.jocher@ultralytics.com>
正在显示
请
注册
或者
登录
后发表评论