sp-uhh / sgmse

Score-based Generative Models (Diffusion Models) for Speech Enhancement and Dereverberation
MIT License
454 stars 69 forks source link

How to make the SGMSE in causal way #34

Open Leaou-Bin opened 11 months ago

Leaou-Bin commented 11 months ago

Hi, when I reproduce the causality in SGMSE (actual in SGMSE-BBED) as the paper 'SPEECH SIGNAL IMPROVEMENT USING CAUSAL GENERATIVE DIFFUSION MODELS' in ICASSP2023. I make the following changement for score model ncsnpp+ as in paper :

  1. add causal pad for 3*3conv2d.
  2. block the attention module
  3. block the Groupnorm
  4. replace the layerspp.ResnetBlockBigGANpp's downsample_2d, upsample_2d with causal conv2d and TransposedConv2D,block the groupnorm in module
  5. replace the layerspp.Downsample Upsample as above in 4.

But it still not work, could you give me some advice on it, Thanks!