Export Linear

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!