We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2a24b90 commit b04549dCopy full SHA for b04549d
src/pytorch_lightning/core/datamodule.py
@@ -150,7 +150,7 @@ def predict_dataloader():
150
return [dataloader(ds) for ds in predict_dataset]
151
return dataloader(predict_dataset)
152
153
- datamodule = cls()
+ datamodule = cls(batch_size=batch_size)
154
if train_dataset is not None:
155
datamodule.train_dataloader = train_dataloader
156
if val_dataset is not None:
0 commit comments