mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
mems initialized to None in run_transfo
This commit is contained in:
parent
cfcb95417c
commit
43b9af0cac
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user