Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
Fairseq
Commits
557b99d1
Commit
557b99d1
authored
Nov 12, 2017
by
Myle Ott
Browse files
Fix Flake8
parent
1b42c8c4
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
5 deletions
+1
-5
fairseq/models/lstm.py
fairseq/models/lstm.py
+0
-4
fairseq/progress_bar.py
fairseq/progress_bar.py
+1
-1
No files found.
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
...
...
fairseq/progress_bar.py
View file @
557b99d1
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment