A question for train_action.py

[SoliveTech]

Thanks for your excellent work！ In the picture from train_action.py, I find that the "idx" is got in the for loop. but after the loop The if statement calls the variable "idx " again. Why the "idx " can be called in here?

[Walter0807]

Good catch! You can move it into the loop (pull request is welcomed if you want).