testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add set_device at experiment setup

This commit is contained in:
chenfeiyu 2020-12-11 19:45:49 +08:00
parent a5f1605051
commit a079e767df
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
parakeet/models/waveflow.py
View File

@ -488,7 +488,7 @@ class ConditionalWaveFlow(nn.LayerList):
z, log_det_jacobian = self.decoder(audio, condition) z, log_det_jacobian = self.decoder(audio, condition)
return z, log_det_jacobian return z, log_det_jacobian
@paddle.fluid.dygraph.no_grad @paddle.no_grad()
def synthesize(self, mel): def synthesize(self, mel):
condition = self.encoder(mel, trim_conv_artifact=True) #(B, C, T) condition = self.encoder(mel, trim_conv_artifact=True) #(B, C, T)
batch_size, _, time_steps = condition.shape batch_size, _, time_steps = condition.shape

1
parakeet/training/experiment.py
View File

@ -59,6 +59,7 @@ class ExperimentBase(object):
self.args = args self.args = args
def setup(self): def setup(self):
paddle.set_device(self.args.device)
if self.parallel: if self.parallel:
self.init_parallel() self.init_parallel()