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

Commit 0be8642

Browse files
ranjita-naikebrevdo
authored andcommitted
Fixing #405
PiperOrigin-RevId: 233662426
1 parent c42f976 commit 0be8642

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nmt/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -437,7 +437,7 @@ def get_model_creator(hparams):
437437
if (hparams.encoder_type == "gnmt" or
438438
hparams.attention_architecture in ["gnmt", "gnmt_v2"]):
439439
model_creator = gnmt_model.GNMTModel
440-
elif hparams.attention_architecture == "standard":
440+
elif hparams.attention and hparams.attention_architecture == "standard":
441441
model_creator = attention_model.AttentionModel
442442
elif not hparams.attention:
443443
model_creator = nmt_model.Model

0 commit comments

Comments
 (0)