Commit 871be389 authored by Myle Ott's avatar Myle Ott
Browse files

Faster fconv generation

parent 0e8414f9
...@@ -279,8 +279,11 @@ class FConvDecoder(FairseqIncrementalDecoder): ...@@ -279,8 +279,11 @@ class FConvDecoder(FairseqIncrementalDecoder):
encoder_a, encoder_b = self._split_encoder_out(encoder_out, incremental_state) encoder_a, encoder_b = self._split_encoder_out(encoder_out, incremental_state)
# embed tokens and combine with positional embeddings # embed tokens and combine with positional embeddings
pos_embed = self.embed_positions(prev_output_tokens, incremental_state)
if incremental_state is not None:
prev_output_tokens = prev_output_tokens[:, -1:]
x = self._embed_tokens(prev_output_tokens, incremental_state) x = self._embed_tokens(prev_output_tokens, incremental_state)
x += self.embed_positions(prev_output_tokens, incremental_state) x += pos_embed
x = F.dropout(x, p=self.dropout, training=self.training) x = F.dropout(x, p=self.dropout, training=self.training)
target_embedding = x target_embedding = x
......
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