update params

2022-06-19 20:47:06 +07:00 · 2022-06-19 20:47:06 +07:00 · 02ead8c05c
parent ff8b0533ac
commit 02ead8c05c
1 changed files with 1 additions and 1 deletions
--- a/codes/models/audio/music/unet_diffusion_waveform_gen3.py
+++ b/codes/models/audio/music/unet_diffusion_waveform_gen3.py
@ -312,7 +312,7 @@ class DiffusionWaveformGen(nn.Module):
        groups = {
            'input_blocks': list(self.input_blocks.parameters()),
            'output_blocks': list(self.output_blocks.parameters()),
-            'middle_transformer': list(self.middle_block.parameters()),
+            'middle_rrdb': list(self.middle_block.parameters()),
        }
        return groups