How to make the SGMSE in causal way

[Leaou-Bin]

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!