Skip to content

Commit

Permalink
change default dataloader batch size
Browse files Browse the repository at this point in the history
  • Loading branch information
afrendeiro committed Mar 12, 2024
1 parent 48af91d commit 176c370
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion wsi_core/WholeSlideImage.py
Original file line number Diff line number Diff line change
Expand Up @@ -427,7 +427,7 @@ def as_tile_bag(self):
)
return dataset

def as_data_loader(self, batch_size: int = 128, with_coords: bool = False, **kwargs):
def as_data_loader(self, batch_size: int = 32, with_coords: bool = False, **kwargs):
from functools import partial
from wsi_core.utils import collate_features
from torch.utils.data import DataLoader
Expand Down

0 comments on commit 176c370

Please sign in to comment.