Unpin thop once onnx import is resolved

drivendataorg / zamba

A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos

https://zamba.drivendata.org/docs/stable/

MIT License

118 stars 27 forks source link

Closed ejm714 closed 2 months ago

ejm714 commented 2 years ago

In https://github.com/drivendataorg/zamba/pull/191, we pin thop to an earlier version since the latest causes an onnx import error. This is currently being discussed in issue https://github.com/Lyken17/pytorch-OpCounter/issues/173#issuecomment-1153194928. Once a new release is cut, we can unpin thop.