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
278 stars 63 forks source link

Update decompositions.py #7

Closed Gaochenyin closed 2 years ago

Gaochenyin commented 3 years ago

partial_tucker() got an unexpected keyword argument 'ranks'