Add support for ViT in timm module. - Githubissues

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework

MIT License

2.83k stars 306 forks source link

Add support for ViT in timm module. #126

Closed zhuoyan-xu closed 1 year ago

zhuoyan-xu commented 1 year ago

Regarding timm ViT support in #124.

resolves https://github.com/sovrasov/flops-counter.pytorch/issues/124