"vscode:/vscode.git/clone" did not exist on "142b69f24b57e5d358edeaa1569955f7684fee93"
Commit 45634f87 authored by Victor SANH's avatar Victor SANH Committed by Lysandre Debut
Browse files

fix Sampler in distributed training - evaluation

parent af1ee9e6
......@@ -264,7 +264,7 @@ def evaluate(args, model, tokenizer, prefix=""):
args.eval_batch_size = args.per_gpu_eval_batch_size * max(1, args.n_gpu)
# Note that DistributedSampler samples randomly
eval_sampler = SequentialSampler(dataset) if args.local_rank == -1 else DistributedSampler(dataset)
eval_sampler = SequentialSampler(dataset)
eval_dataloader = DataLoader(dataset, sampler=eval_sampler, batch_size=args.eval_batch_size)
# Eval!
......
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