Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Remove ComputeLoss from TorchScript graph #483

Open
zhiqwang opened this issue Feb 9, 2023 · 0 comments
Open

Remove ComputeLoss from TorchScript graph #483

zhiqwang opened this issue Feb 9, 2023 · 0 comments
Labels
code quality Code format and unit tests deployment Inference acceleration for production

Comments

@zhiqwang
Copy link
Owner

zhiqwang commented Feb 9, 2023

🚀 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
@zhiqwang zhiqwang added deployment Inference acceleration for production code quality Code format and unit tests labels Feb 9, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
code quality Code format and unit tests deployment Inference acceleration for production
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@zhiqwang