-
由 Jan Hajek 提交于
* 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>