How to do prediction on train and validation dataset?

implementation help
1

Since we fix the train, validation and test while initialization of the data module. How to specifically test/predict the train and validation dataset without creating a new data module?

2

One simple way would be to just pass your dataloaders into the respective trainer method:

dataloader = DataLoader(NewDataset())

trainer.test(dataloader)
# or
trainer.predict(dataloader)

So you don’t rely on the DataModule you use for training.