Does it support torch.einsum for FLOPs?
wztdream opened this issue · comments
wztdream commented
Hi,
I use a lot of torch.einsum in my model, does this repo account it's FLOPs?
Ligeng Zhu commented
Unfornately, it is not supported yet. Einsum is a very flexible operation and it is hard to trace the actual computaion details.