JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡
https://julesbelveze.github.io/bert-squeeze/
78 stars 10 forks source link

Can it be used in the speech enhancement ? #40

Open zuowanbushiwo opened 1 year ago

zuowanbushiwo commented 1 year ago

Hi @JulesBelveze I trained a speech enhancement network using pytorch-lightning, and now I want to do some lightweighting on it, can I use this project? The network structure is Convolution Recurrent Network: image image

thanks! looking forward to your suggestion!

JulesBelveze commented 1 year ago

Hey @zuowanbushiwo thanks for your interest! Even though this repo is mainly designed to handle Transformers-based model I am currently working a significant refactoring. It should allow for more flexibility and be more user-friendly.

My plate is quite full atm and the refactoring got a bit delayed. How about I ping you once it's done and we check together if we can make it work? 🤓

zuowanbushiwo commented 1 year ago

@JulesBelveze Thanks,Looking forward to it, I'll view source files now first.

JulesBelveze commented 1 year ago

Sweet!