Skip to content

Commit

Permalink
Expand worker number for acceleration
Browse files Browse the repository at this point in the history
Signed-off-by: begeekmyfriend <[email protected]>
  • Loading branch information
begeekmyfriend committed Dec 3, 2019
1 parent 7a78e69 commit 35e7f45
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,7 +169,7 @@ def evaluating(model):
def validate(model, criterion, valate_dataset, iteration, collate_fn, distributed_run, args):
"""Handles all the validation scoring and printing"""
with evaluating(model), torch.no_grad():
val_loader = DataLoader(valate_dataset, num_workers=1, shuffle=False,
val_loader = DataLoader(valate_dataset, num_workers=8, shuffle=False,
batch_size=args.batch_size//len(args.validation_anchor_dirs),
pin_memory=False, collate_fn=collate_fn)

Expand Down

0 comments on commit 35e7f45

Please sign in to comment.