Skip to content

Commit b04549d

Browse files
bug fix in from_datasets method
1 parent 2a24b90 commit b04549d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/pytorch_lightning/core/datamodule.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def predict_dataloader():
150150
return [dataloader(ds) for ds in predict_dataset]
151151
return dataloader(predict_dataset)
152152

153-
datamodule = cls()
153+
datamodule = cls(batch_size=batch_size)
154154
if train_dataset is not None:
155155
datamodule.train_dataloader = train_dataloader
156156
if val_dataset is not None:

0 commit comments

Comments
 (0)