early stopping or checkpointing without val step
Is your feature request related to a problem? Please describe.
When val step isn't defined, early stopping nor checkpointing work.
Describe the solution you'd like
Early stopping should look at training_step metrics instead