Skip to content

Commit

Permalink
add multi-card for run_eval dataloader
Browse files Browse the repository at this point in the history
  • Loading branch information
Mark-ZhouWX committed Jun 12, 2023
1 parent b502686 commit c90c667
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,17 +165,17 @@ def train(args):
augment=False,
rect=args.rect,
single_cls=args.single_cls,
batch_size=args.per_batch_size * 2,
batch_size=args.per_batch_size,
stride=max(args.network.stride),
)
eval_dataloader = create_loader(
dataset=eval_dataset,
batch_collate_fn=eval_dataset.test_collate_fn,
dataset_column_names=eval_dataset.dataset_column_names,
batch_size=args.per_batch_size * 2,
batch_size=args.per_batch_size,
epoch_size=1,
rank=0,
rank_size=1,
rank=args.rank,
rank_size=args.rank_size,
shuffle=False,
drop_remainder=False,
num_parallel_workers=args.data.num_parallel_workers,
Expand Down

0 comments on commit c90c667

Please sign in to comment.