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) (#1502)

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 authored Oct 5, 2023
1 parent 9727229 commit a762a4d
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 a762a4d

Please sign in to comment.