Auto-scale batch size triggers "on_train_end"
When running Trainer with auto_scale_batch_size set to true, the on_train_end function is called at the start, but never at the end, because there is a check to not run teardown multiple times.