/tensor_train_rnn

Replace some weights matrices of linear layers of RNNs with matrices in tensor train format. Get better generalization and a lot less parameters.

Primary LanguageJupyter Notebook

Stargazers