search

huggingface / pytorch_block_sparse

Fast Block Sparse Matrices for Pytorch
Other
545 stars 35 forks source link

Specify the sparsity for the BlockSparseLinear #10

Closed zheng-ningxin closed 3 years ago

zheng-ningxin commented 4 years ago

Hi! thanks for the excellent job! I am trying to use pytorch_block_sparse to do some research work and have some questions. Can we specify the sparsity for the BlockSparseLinear layer? Looking forward to hearing from you~

zheng-ningxin commented 3 years ago

Solved.