Skip to content

Commit

Permalink
fix valid
Browse files Browse the repository at this point in the history
  • Loading branch information
ilyes319 committed Jun 6, 2024
1 parent 5463656 commit 122a50c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mace/cli/run_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -434,7 +434,7 @@ def main() -> None:
dataset=valid_set,
batch_size=args.valid_batch_size,
sampler=valid_samplers[head] if args.distributed else None,
shuffle=False
shuffle=False,
drop_last=False,
pin_memory=args.pin_memory,
num_workers=args.num_workers,
Expand Down

0 comments on commit 122a50c

Please sign in to comment.