update params
This commit is contained in:
parent
ff8b0533ac
commit
02ead8c05c
|
@ -312,7 +312,7 @@ class DiffusionWaveformGen(nn.Module):
|
||||||
groups = {
|
groups = {
|
||||||
'input_blocks': list(self.input_blocks.parameters()),
|
'input_blocks': list(self.input_blocks.parameters()),
|
||||||
'output_blocks': list(self.output_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
|
return groups
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user