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
-
🐛 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