search

zhiqwang / yolort

yolort is a runtime stack for yolov5 on specialized accelerators such as tensorrt, libtorch, onnxruntime, tvm and ncnn.
https://zhiqwang.com/yolort
GNU General Public License v3.0
717 stars 153 forks source link

Remove `ComputeLoss` from TorchScript graph #483

Open zhiqwang opened 1 year ago

zhiqwang commented 1 year ago

🚀 The feature

Remove following ComputeLoss when building TorchScript graph https://github.com/zhiqwang/yolov5-rt-stack/blob/3eab7bee28ef23cd86012fe5d5a309a6dad7d56c/yolort/models/box_head.py#L85

Motivation, pitch

More deployment friendly.

Alternatives

No response

Additional context

No response