search

youweiliang / evit

Python code for ICLR 2022 spotlight paper EViT: Expediting Vision Transformers via Token Reorganizations
Apache License 2.0
162 stars 19 forks source link

How to handle "compute MAC error"? #8

Open pzSuen opened 2 years ago

pzSuen commented 2 years ago

Thanks for your code.

I test my own model by the code you provide in speed_test.py, but I encounter this problem.

https://github.com/youweiliang/evit/blob/97e58f610c51d4b74a070341739e41647dced32c/speed_test.py#L106

What's the reason for it? How to deal with it?

youweiliang commented 2 years ago

Hi, you can add print(e) to see the error message.