search

jacobgil / pytorch-tensor-decompositions

PyTorch implementation of [1412.6553] and [1511.06530] tensor decomposition methods for convolutional layers.
https://jacobgil.github.io/deeplearning/tensor-decompositions-deep-learning
279 stars 63 forks source link

CP_decompose error #11

Open sxdnbnb opened 1 year ago

sxdnbnb commented 1 year ago

I use pytorch 1.10 and tensorly 0.4.0.,but I got error as following: File "/content/drive/MyDrive/Tensor_decomposing/decompositions.py", line 14, in cp_decomposition_conv_layer parafac(layer.weight.data, rank=rank, init='svd') File "/usr/local/lib/python3.9/dist-packages/tensorly/decomposition/candecomp_parafac.py", line 153, in parafac factor = T.transpose(T.solve(T.transpose(pseudo_inverse), T.transpose(factor))) File "/usr/local/lib/python3.9/dist-packages/tensorly/backend/pytorchbackend.py", line 152, in solve solution, = torch.gesv(matrix2, matrix1) AttributeError: module 'torch' has no attribute 'gesv'

siyoooooo commented 1 year ago

Did you fix this? I'm also having the same problem

sxdnbnb commented 1 year ago

Did you fix this? I'm also having the same problem

No