Browse Source

更新 'svoice/models/swave.py'

master
lwj 1 month ago
parent
commit
e9b3e52e31
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      svoice/models/swave.py

+ 3
- 3
svoice/models/swave.py View File

@@ -14,7 +14,7 @@ import mindspore.nn as nn
import numpy as np
from mindspore.common.initializer import One
from mindspore import Parameter, context, Tensor, ms_function
from mindspore.common.initializer import initializer, XavierUniform, Uniform, HeNormal
from mindspore.common.initializer import initializer, XavierUniform, Uniform, HeNormal, Normal
#from svoice.utils import overlap_and_add
# from ..utils import capture_init
# from svoice.utils import capture_init
@@ -294,8 +294,8 @@ class SWave(nn.Cell):
for p in self.get_parameters():
if p.dim() > 1:
# initializer(XavierUniform(), p.shape, mindspore.float32)
mindspore.common.initializer.XavierUniform(p)
# initializer(HeNormal(), p.shape, mindspore.float32)
# mindspore.common.initializer.XavierUniform(p)
initializer(Normal(), p.shape, mindspore.float32)

def construct(self, mixture):
# self.capture_init(self.__init__)


Loading…
Cancel
Save