Does it support torch.einsum for FLOPs?

Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

MIT License

4.85k stars 528 forks source link

Closed wztdream closed 2 years ago

wztdream commented 2 years ago

Hi, I use a lot of torch.einsum in my model, does this repo account it's FLOPs?

Lyken17 commented 2 years ago

Unfornately, it is not supported yet. Einsum is a very flexible operation and it is hard to trace the actual computaion details.