error place for scheduler.step()

[lx709]

In line 100 of main_mae.py, scheduler.step() should be placed out of the inner for-loop.