Skip to content
This repository was archived by the owner on Aug 1, 2023. It is now read-only.

Commit 17c1f47

Browse files
jhcrossfacebook-github-bot
authored andcommitted
model export: remove extraneous model.eval()
Differential Revision: D13969501 fbshipit-source-id: 525464bd46f7d5c925e2392ba93f6ef6533dfb63
1 parent d23a172 commit 17c1f47

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

pytorch_translate/ensemble_export.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -267,9 +267,6 @@ def forward(self, src_tokens, src_lengths):
267267
torch.jit._fork(model.encoder, src_tokens_seq_first, src_lengths)
268268
)
269269

270-
# evaluation mode
271-
model.eval()
272-
273270
for i, (model, future) in enumerate(zip(self.models, futures)):
274271
if isinstance(model.encoder, TransformerEncoder):
275272
encoder_out = future
@@ -1361,9 +1358,6 @@ def forward(self, src_tokens, src_lengths, char_inds, word_lengths):
13611358
src_tokens_seq_first, src_lengths, char_inds, word_lengths
13621359
)
13631360

1364-
# evaluation mode
1365-
model.eval()
1366-
13671361
# "primary" encoder output (vector representations per source token)
13681362
encoder_outputs = encoder_out[0]
13691363
outputs.append(encoder_outputs)

0 commit comments

Comments
 (0)