mirror of
https://github.com/wassname/TTS.git
synced 2026-10-29 12:51:00 +08:00
glow-tts fix for saving inverse weight
This commit is contained in:
1 parent
3d30dae8f3
commit
5c87753e88
1 file changed
+2
-1
@@ -128,8 +128,9 @@ class InvConvNear(nn.Module):
|
||||
return z, logdet
|
||||
|
||||
def store_inverse(self):
|
||||
self.weight_inv = torch.inverse(
|
||||
weight_inv = torch.inverse(
|
||||
self.weight.float()).to(dtype=self.weight.dtype)
|
||||
self.weight_inv = nn.Parameter(weight_inv, requires_grad=False)
|
||||
|
||||
|
||||
class CouplingBlock(nn.Module):
|
||||
|
||||
Reference in new issue
Block a user