I just fixed the problem about backward

[LoveDLWujing]

here is the solution
#2 (comment)

[wallcuber]

The code of @Vishwanath1999 seems worked！Thank you, buddy!

[wallcuber]

But the different between old actions and actions from mu_states is still not clear to me. Maybe I need more studying.