Question

[JackHeroFly]

Hello,thanks for your code. When I use the non local module in my model, I am always reminded that my graphics card is out of memory. In fact, when the module is removed, the model can be trained normally. Does this module need to occupy a lot of memory?