Skip to content

Commit 720a0b5

Browse files
authored
Default dataloaders to list (#137)
1 parent 2081cbd commit 720a0b5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lighter/system.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def __init__(
109109
self.inferer = inferer
110110

111111
# Bypasses LightningModule's check for dataloader and step methods. We define them dynamically in self.setup().
112-
self.train_dataloader = self.val_dataloader = self.test_dataloader = self.predict_dataloader = lambda: None
112+
self.train_dataloader = self.val_dataloader = self.test_dataloader = self.predict_dataloader = lambda: []
113113
self.training_step = self.validation_step = self.test_step = self.predict_step = lambda: None
114114

115115
def forward(self, input: Union[Tensor, List[Tensor], Tuple[Tensor], Dict[str, Tensor]]) -> Any:

0 commit comments

Comments
 (0)