get current stage in pytorch-lightning

[2533245542]

Hi everyone,

I have a question related to pytorch lightning's module.

Say I am in the forward function and I want to see if I am in which of the train/val/test stage.

Is there a way to do so?

Didn't get lucky with google search and a look into the source code seems like self.training is boolean for indicating training/non-training stage, but still not enough.

[rohitgr7]

use self.trainer.training/testing/validating