Replies: 1 comment
-
Dear @its-dron, It is still an experimental feature and will change. It is why it hasn t been added to the doc yet. But you can add a predict_dataloader to your DataModule and it should work. It will call the predict function of the Lightning Module you can override. This will likely be renamed predict_step soon Best, |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
🐛 Bug
Unable to use the new
Trainer.predict
method with datamodule input. The following error is thrown:AttributeError: 'Trainer' object has no attribute 'predict_dataloaders'
Best I can tell, the error originates from this line: https://github.com/PyTorchLightning/pytorch-lightning/blob/6cc1a060785448eacd016f6689d592ac822adaaf/pytorch_lightning/trainer/predict_loop.py#L33
Additionally, the documentation is vague about which dataloader is used from the datamodule.
To Reproduce
https://colab.research.google.com/drive/1ZD5xwVifXnxkbAypyeP-nnAXa1IYCBNx?usp=sharing
Expected behavior
Don't throw an error?
Environment
conda
,pip
, source):Additional context
Beta Was this translation helpful? Give feedback.
All reactions