Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
Fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
@ethanwharris
ethanwharris committed Aug 24, 2021
1 parent c5152e1 commit 742a156
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions flash/core/data/data_pipeline.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -416,7 +416,7 @@ def _attach_preprocess_to_model(
if not was_seq:
dataloader = dataloader[0]

self._patch_dataloader(model, dataloader, stage)
dataloader = self._patch_dataloader(model, dataloader, stage)

self._set_loader(model, whole_attr_name, dataloader)

Expand Down Expand Up @@ -545,7 +545,7 @@ def _detach_preprocessing_from_model(self, model: "Task", stage: Optional[Runnin
if not was_seq:
dataloader = dataloader[0]

self._patch_dataloader(model, dataloader, stage)
dataloader = self._patch_dataloader(model, dataloader, stage)

self._set_loader(model, whole_attr_name, dataloader)

Expand Down

0 comments on commit 742a156

Please sign in to comment.