JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

Home Page:https://julesbelveze.github.io/bert-squeeze/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can it be used in the speech enhancement ?

zuowanbushiwo opened this issue · comments

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!

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? 🤓

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

Sweet!