Necolizer
Results
2
issues of
Necolizer
Thank you for your excellent work! This issue #40 gives a toy example of multi-head self-attention implemented using `torch.einsum`. Running that script gets an AttributeError raised in function `_einsum_flops_compute()`. To...
- Version: python==3.7.9, torch==1.9.0+cu111, torchvision==0.10.0+cu111, calflops==0.2.0 - Problem: Here's an example to see the error: ```python import torch import torch.nn as nn from calflops import calculate_flops class Model(nn.Module): def __init__(self,...