Somehow I am unable to debug the training_step when I use the VSCode debugger. It just doesn’t stop at the breakpoint. However, for the validation_step it works. And I am also not able to print anything inside the training_step function even with flush=True.
What could be the reason for this?