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.

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

Suppress ONNX export trace warning #5437

Merged
merged 1 commit into from
Nov 5, 2021

Commits on Nov 5, 2021

  1. Update yolo.py

    Checking for `onnx_dynamic` first should suppress the warning:
    
    ```log
    TracerWarning: Converting a tensor to a Python boolean might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs!
      if self.grid[i].shape[2:4] != x[i].shape[2:4] or self.onnx_dynamic
    ```
    deepsworld authored and glenn-jocher committed Nov 5, 2021
    Configuration menu
    Copy the full SHA
    c8571c7 View commit details
    Browse the repository at this point in the history