search

facebookresearch / ToMe

A method to increase the speed and lower the memory footprint of existing vision transformers.
Other
931 stars 67 forks source link

Does ToMe work for Restormer? #27

Closed tttt-0814 closed 1 year ago

tttt-0814 commented 1 year ago

Can I apply ToMe for Restormer ? (https://github.com/swz30/Restormer)

dbolya commented 1 year ago

You could probably do something similar to tomesd since denoising/deblurring etc requires all the tokens to be in tact. But, it would probably require some extra effort to get working.

tttt-0814 commented 1 year ago

I got it. Thanks.