Fix error AttributeError: 'RNNModel' object has no attribute 'model_type' (pytorch#614)

Author: joistick11

word_language_model/generate.py

@@ -49,14 +49,16 @@
 
 corpus = data.Corpus(args.data)
 ntokens = len(corpus.dictionary)
-if model.model_type != 'Transformer':
+
+is_transformer_model = hasattr(model, 'model_type') and model.model_type == 'Transformer'
+if not is_transformer_model:
     hidden = model.init_hidden(1)
 input = torch.randint(ntokens, (1, 1), dtype=torch.long).to(device)
 
 with open(args.outf, 'w') as outf:
     with torch.no_grad():  # no tracking history
         for i in range(args.words):
-            if model.model_type == 'Transformer':
+            if is_transformer_model:
                 output = model(input, False)
                 word_weights = output[-1].squeeze().div(args.temperature).exp().cpu()
                 word_idx = torch.multinomial(word_weights, 1)[0]