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

Fix Flake8

parent 1b42c8c4
...@@ -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
......
...@@ -54,7 +54,7 @@ class progress_bar(object): ...@@ -54,7 +54,7 @@ class progress_bar(object):
def _str_pipes(self, stats): def _str_pipes(self, stats):
return ' | '.join(key + ' ' + stats[key].strip() return ' | '.join(key + ' ' + stats[key].strip()
for key in stats.keys()) for key in stats.keys())
def _format_stats(self, stats): def _format_stats(self, stats):
postfix = OrderedDict(stats) postfix = OrderedDict(stats)
......
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