Skip to content

GitLab

Commit 557b99d1 authored by Myle Ott's avatar Myle Ott
Browse files

Fix Flake8

parent 1b42c8c4
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
fairseq/models/lstm.py
View file @ 557b99d1
...@@ -126,7 +126,6 @@ class LSTMDecoder(FairseqIncrementalDecoder): ...@@ -126,7 +126,6 @@ class LSTMDecoder(FairseqIncrementalDecoder):
def forward(self, input_tokens, encoder_out): def forward(self, input_tokens, encoder_out):
bsz, seqlen = input_tokens.size() bsz, seqlen = input_tokens.size()
num_layers = len(self.layers)
# get outputs from encoder # get outputs from encoder
encoder_outs, _, _ = encoder_out encoder_outs, _, _ = encoder_out
...@@ -220,10 +219,7 @@ class LSTMDecoder(FairseqIncrementalDecoder): ...@@ -220,10 +219,7 @@ class LSTMDecoder(FairseqIncrementalDecoder):
def _init_prev_states(self, input_tokens, encoder_out): def _init_prev_states(self, input_tokens, encoder_out):
_, encoder_hiddens, encoder_cells = encoder_out _, encoder_hiddens, encoder_cells = encoder_out
bsz = input_tokens.size(0)
num_layers = len(self.layers) num_layers = len(self.layers)
embed_dim = encoder_hiddens.size(2)
prev_hiddens = [encoder_hiddens[i] for i in range(num_layers)] prev_hiddens = [encoder_hiddens[i] for i in range(num_layers)]
prev_cells = [encoder_cells[i] for i in range(num_layers)] prev_cells = [encoder_cells[i] for i in range(num_layers)]
return prev_hiddens, prev_cells return prev_hiddens, prev_cells
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment