Generate was broken for QRNN as QRNN didn't flush their previously st…

…ored X, resulting in diff batch sizes, hence angry concat dimensions

generate.py

@@ -17,6 +17,8 @@
 # Model parameters.
 parser.add_argument('--data', type=str, default='./data/penn',
  help='location of the data corpus')
+parser.add_argument('--model', type=str, default='LSTM',
+ help='type of recurrent net (LSTM, QRNN)')
 parser.add_argument('--checkpoint', type=str, default='./model.pt',
  help='model checkpoint to use')
 parser.add_argument('--outf', type=str, default='generated.txt',
@@ -47,6 +49,8 @@
 with open(args.checkpoint, 'rb') as f:
  model = torch.load(f)
 model.eval()
+if args.model == 'QRNN':
+ model.reset()
 
 if args.cuda:
  model.cuda()