Skip the train_step based on batch statistics
I have an issue where I want to skip a training step based on the input batch statistics. I am just wondering if it is possible to skip the train_step in a lightning model i.e. the training would just ignore this batch and carry on with the next one?
usually, I would make sure the batches are always generated in an acceptable manner but I have a hard to track bug which happens rarely and due to some deadline pressures, I would still like to continue the training by skipping this batch. Is there a way to do this with the lightning framework?