Commit 43b9af0c authored by thomwolf's avatar thomwolf
Browse files

mems initialized to None in run_transfo

parent cfcb9541
......@@ -100,7 +100,7 @@ def main():
total_len, total_loss = 0, 0.
start_time = time.time()
with torch.no_grad():
mems = tuple()
mems = None
for idx, (data, target, seq_len) in enumerate(eval_iter):
ret = model(data, target, *mems)
loss, mems = ret
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment