Fixing #405

PiperOrigin-RevId: 233662426
tensorflow · Feb 13, 2019 · 0be8642 · 0be8642
1 parent c42f976
commit 0be8642
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nmt/train.py b/nmt/train.py
@@ -437,7 +437,7 @@ def get_model_creator(hparams):
   if (hparams.encoder_type == "gnmt" or
       hparams.attention_architecture in ["gnmt", "gnmt_v2"]):
     model_creator = gnmt_model.GNMTModel
-  elif hparams.attention_architecture == "standard":
+  elif hparams.attention and hparams.attention_architecture == "standard":
     model_creator = attention_model.AttentionModel
   elif not hparams.attention:
     model_creator = nmt_model.Model