Fix observations on PPO trainer (#340)

* Fix observations on PPO trainer * tested and fixed the fix
Unity-Technologies · Feb 15, 2018 · ef61887 · ef61887
1 parent 39df903
commit ef61887
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ppo/trainer.py b/python/ppo/trainer.py
@@ -122,7 +122,7 @@ def process_experiences(self, info, time_horizon, gamma, lambd):
                 else:
                     feed_dict = {self.model.batch_size: len(info.states)}
                     if self.use_observations:
-                        for i in range(self.info.observations):
+                        for i in range(len(info.observations)):
                             feed_dict[self.model.observation_in[i]] = info.observations[i]
                     if self.use_states:
                         feed_dict[self.model.state_in] = info.states