fixing transfo eval script

examples/eval_transfo_xl.py

@@ -111,7 +111,7 @@ def evaluate(eval_iter):
         mems = tuple()
         for idx, (data, target, seq_len) in enumerate(eval_iter):
             ret = model(data, target, *mems)
-            loss, mems = ret[0], ret[1:]
+            loss, mems = ret
             loss = loss.mean()
             total_loss += seq_len * loss.item()
             total_len += seq_len

pytorch_pretrained_bert/modeling_transfo_xl.py

@@ -1215,7 +1215,8 @@ class TransfoXLModel(TransfoXLPreTrainedModel):
         # So, have to initialize size(0) mems inside the model forward.
         # Moreover, have to return new_mems to allow nn.DataParallel to piece
         # them together.
-        if not mems: mems = self.init_mems(data)
+        if not mems:
+            mems = self.init_mems(data)
 
         hidden, new_mems = self._forward(data, mems=mems)
         if target is None: