Does this library work with einops?
vedantroy opened this issue · comments
If I use the einops package (https://github.com/arogozhnikov/einops), will this library still be able to accurately the # of flops in my model?
I am afraid not. Einops is not a Module-level operation and currently it is not traced by thop.
I am working on some alternatives using onnx or torch.fx to trace such a behavior.
Got it. Thanks for the response!