use of  pdb.set_trace() inside LightningModule
So being new to lightning, I have rather a basic question that I couldn't quite find the answer to in the docs.  So in regular pytorch I can use python's pdb module to pause the execution at any point and like check tensor dimensions/values, etc :
from pdb import set_trace
set_trace()
However I seem not be able to do so with lightning, whether in forward() or training_step() or any other method in the lightning module I can pause the execution at any point (obviously without having to go into lightning's internal modules). Is this supposed to be not possible or what am i missing?