optimize: 精简未用到的配置项并在特征提取初步引入mps (#32)

This commit is contained in:
源文雨
2023-04-11 18:14:55 +08:00
committed by GitHub
parent 0656591373
commit ecc744d748
10 changed files with 82 additions and 57 deletions

View File

@@ -345,7 +345,7 @@ class SourceModuleHnNSF(torch.nn.Module):
def forward(self, x,upp=None):
sine_wavs, uv, _ = self.l_sin_gen(x,upp)
if(self.is_half==True):sine_wavs=sine_wavs.half()
if(self.is_half):sine_wavs=sine_wavs.half()
sine_merge = self.l_tanh(self.l_linear(sine_wavs))
return sine_merge,None,None# noise, uv
class GeneratorNSF(torch.nn.Module):