Jan Hajek 7a0a81fd1d
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>
2021-03-06 12:02:10 -08:00
..
2021-02-01 13:51:11 -08:00
2021-03-01 17:13:34 -08:00
2021-03-06 12:02:10 -08:00
2021-03-02 23:08:21 -08:00
2021-01-04 19:54:09 -08:00
2021-01-04 19:54:09 -08:00
2021-01-04 19:54:09 -08:00
2021-01-04 19:54:09 -08:00