BUG：training_module.py

[mingsyang126]

[Feng0w0]

Please provide a more detailed description of the bug. If you mean there is a problem with the if-else statement, here are using the for-else syntax.