fixed some bugs for main_training_reward_based_new.py
Merge branch 'MotionPerception/wip' of https://github.com/bernhardkaplan/OculomotorControl into MotionPerception/wip