Bug in Equivariant

[KunhaoZhong]

Hi,

There seems to be some issue with the Equivariant setting. First, there's no update_psi_equivariant in the current version. After fixing that, I had a bug that said some tensor was not sent to GPU. Is there another version that equivariant is tested? Or any suggestions on how to solve it? Thanks a lot!

[MuawizChaudhary]

Hi Kunhao,

Thanks for the interest. Can you verify with me what version of kymatio you have? Unfortunately, due to an update in Kymatio, which slightly changed the dictionary structure of the psi and phi, this error of tensor not being on GPU occurs.

If you can contact me on muawiz.chaudhary@mila.quebec I can perhaps give you corrected code.