Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
Newbeeer committed May 28, 2023
1 parent acad8da commit 8149b85
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fid.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ def calculate_inception_stats(
# TODO: test different seeds and take the minimum
all_batches = torch.arange(len(dataset_obj)).tensor_split(num_batches)
rank_batches = all_batches[dist.get_rank() :: dist.get_world_size()]
data_loader = torch.utils.data.DataLoader(dataset_obj, batch_sampler=rank_batches, num_workers=num_workers, prefetch_factor=prefetch_factor)
data_loader = torch.utils.data.DataLoader(dataset_obj, batch_sampler=rank_batches, num_workers=0, prefetch_factor=prefetch_factor)

# Accumulate statistics.
dist.print0(f'Calculating statistics for {len(dataset_obj)} images...')
Expand Down

0 comments on commit 8149b85

Please sign in to comment.