search

QUVA-Lab / e2cnn

E(2)-Equivariant CNNs Library for Pytorch
https://quva-lab.github.io/e2cnn/
Other
596 stars 75 forks source link

Export Linear #77

Open theyearis1981 opened 1 year ago

theyearis1981 commented 1 year ago

Hi!

I am trying to export a network containing linear layers, but I'm getting the following error:

File "X/site-packages/escnn/nn/modules/linear.py", line 305, in export linear.weight.data[:] = self TypeError: can't assign a Linear to a torch.FloatTensor

Do you know what's causing it?

Thanks!

Gabri95 commented 1 year ago

Hi @theyearis1981

I think you posted this on the wrong library, can you move this issue on the newer escnn library?

Gabriele

theyearis1981 commented 1 year ago

Sorry! Just reposted it there.