Ошибка при запуске обученой модели на ML-Agents

При запуске спамит вот этой ошибкой:

mismatch between vector observation size (444) and number of observations written (234) UnityEngine.Debug:AssertFormat (bool,string,object[]) Unity.MLAgents.Inference.VectorObservationGenerator:Generate (Unity.MLAgents.Inference.TensorProxy,int,System.Collections.Generic.IEnumerable1<Unity.MLAgents.Inference.AgentInfoSensorsPair>) (at C:/Users/Danil/Desktop/ml-agents-release_2/com.unity.ml-agents/Runtime/Inference/GeneratorImpl.cs:122) Unity.MLAgents.Inference.TensorGenerator:GenerateTensors (System.Collections.Generic.IEnumerable1<Unity.MLAgents.Inference.TensorProxy>,int,System.Collections.Generic.IEnumerable`1<Unity.MLAgents.Inference.AgentInfoSensorsPair>) (at C:/Users/Danil/Desktop/ml-agents-release_2/com.unity.ml-agents/Runtime/Inference/TensorGenerator.cs:141) Unity.MLAgents.Inference.ModelRunner:DecideBatch () (at C:/Users/Danil/Desktop/ml-agents-release_2/com.unity.ml-agents/Runtime/Inference/ModelRunner.cs:165) Unity.MLAgents.Policies.BarracudaPolicy:DecideAction () (at C:/Users/Danil/Desktop/ml-agents-release_2/com.unity.ml-agents/Runtime/Policies/BarracudaPolicy.cs:60) Unity.MLAgents.Agent:DecideAction () (at C:/Users/Danil/Desktop/ml-agents-release_2/com.unity.ml-agents/Runtime/Agent.cs:1173) Unity.MLAgents.Academy:EnvironmentStep () (at C:/Users/Danil/Desktop/ml-agents-release_2/com.unity.ml-agents/Runtime/Academy.cs:508) Unity.MLAgents.AcademyFixedUpdateStepper:FixedUpdate () (at C:/Users/Danil/Desktop/ml-agents-release_2/com.unity.ml-agents/Runtime/Academy.cs:34)

Подскажите что делать. Я новичёк в этом поэтому пожалуйста объясните всё очень просто. Заранее спасибо.