search

YifanXu74 / Evo-ViT

Official implement of Evo-ViT: Slow-Fast Token Evolution for Dynamic Vision Transformer
MIT License
68 stars 5 forks source link

Environment issue #7

Open King4819 opened 9 months ago

King4819 commented 9 months ago

Awesome work ! But I can not run the project correctly yet. Please provide me some information, thanks !

image
YifanXu74 commented 9 months ago

Hello!

This error is commonly due to distributed training. 'model.module.xxx' is only available with distributed training. If you run the training script on one single device, just use model.xxx :)

King4819 commented 9 months ago

@YifanXu74 Thanks, seems like the problem is solved. I have recently studied token-pruning methods, and this is an awesome work !!

YifanXu74 commented 9 months ago

Thanks for your recognition!