Relax on_train_batch_* hook check with dataloader_iter to a warning

[erhoo82]

What does this PR do?

This changes the check of using on_train_batch_start and on_train_batch_end hooks with dataloader_iter from assert to warn.

When these hooks are used with dataloader_iter, the batch_idx argument would not match with the training index depending on how often the batch iteration is called per training step. So, this PR intends to raise warning on such behavior and let users use the training batch hooks for extensive use of dataloader_iter.

Does your PR introduce any breaking changes? If yes, please list them.

Before submitting

• Was this discussed/approved via a GitHub issue? (not for typos and docs)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure your PR does only one thing, instead of bundling different changes together?
• Did you make sure to update the documentation with your changes? (if necessary)
• Did you write any new necessary tests? (not for typos and docs)
• Did you verify new and existing tests pass locally with your changes?
• Did you list all the breaking changes introduced by this pull request?
• Did you update the CHANGELOG? (not for typos, docs, test updates, or minor internal changes/refactors)

PR review

Anyone in the community is welcome to review the PR.
Before you start reviewing, make sure you have read the review guidelines. In short, see the following bullet-list:

• Is this pull request ready for review? (if not, please submit in draft mode)
• Check that all items from Before submitting are resolved
• Make sure the title is self-explanatory and the description concisely explains the PR
• Add labels and milestones (and optionally projects) to the PR so it can be classified

Did you have fun?

Make sure you had fun coding 🙃

[erhoo82]

@carmocca
Please help review and merge.