Question how to reload scene with ml agents

Hi, I am using trained agents in Walker project. When game is over player can reload the entire scene and start from beginning. The problem is that after reloading scene via SceneManager.LoadScene(), the character seems untrained. That happens only after reloading scene. First time everything is ok. How to deal with it?There is no error reported after the scene reset