Skip to content

Commit

Permalink
A call to _process_collate_fn_params was missing in dataloaders.get. (#…
Browse files Browse the repository at this point in the history
…1501)

Otherwise string collate FN would not be instantiated properly IF dataloader name is None and dataset name is present in dataloader params

(cherry picked from commit 8637ee7)
  • Loading branch information
BloodAxe committed Oct 5, 2023
1 parent 9727229 commit 82f26f8
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/super_gradients/training/dataloaders/dataloaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -886,6 +886,8 @@ def get(name: str = None, dataset_params: Dict = None, dataloader_params: Dict =

if dataset is not None:
dataloader_params = _process_sampler_params(dataloader_params, dataset, {})
dataloader_params = _process_collate_fn_params(dataloader_params)

dataloader = DataLoader(dataset=dataset, **dataloader_params)

dataloader.dataloader_params = dataloader_params
Expand Down

0 comments on commit 82f26f8

Please sign in to comment.