dan
/
tacotron2
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
Browse Source

Merge pull request #303 from NTT123/fix-batch-size-1 

fix error when batch size = 1
master
Rafael Valle 4 years ago
committed by GitHub
parent
ca5a22a71a 14dbc37497
commit
2583315739
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      model.py

+ 1
- 1
model.py View File

@ -407,7 +407,7 @@ class Decoder(nn.Module):
mel_output, gate_output, attention_weights = self.decode(
decoder_input)
mel_outputs += [mel_output.squeeze(1)]
gate_outputs += [gate_output.squeeze()]
gate_outputs += [gate_output.squeeze(1)]
alignments += [attention_weights]
mel_outputs, gate_outputs, alignments = self.parse_decoder_outputs(

Write Preview
Loading…
Cancel
Save