Plain Performer, if you are working with say images or other modalities

lucidrains / performer-pytorch

An implementation of Performer, a linear attention-based transformer, in Pytorch

MIT License

1.08k stars 141 forks source link

Plain Performer, if you are working with say images or other modalities #49

Open haoshuai714 opened 3 years ago

haoshuai714 commented 3 years ago

Plain Performer, if you are working with say images or other modalities! ERROR:ModuleNotFoundError: No module named 'local_attention' where is the local_attention module?

lucidrains commented 3 years ago

@haoshuai714 that's weird, it should be taken care off by the dependencies when you pip installed performer. Try pip install local-attention