Merge pull request #200 from keras-rl/fix-tensorboard-support

Fix Tensorboard Compatibility
keras-rl · Apr 30, 2018 · ea7218b · ea7218b
2 parents fac1e61 + cce5b63
commit ea7218b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/rl/core.py b/rl/core.py
@@ -114,8 +114,8 @@ def fit(self, env, nb_steps, action_repetition=1, callbacks=None, verbose=1,
         self._on_train_begin()
         callbacks.on_train_begin()
 
-        episode = 0
-        self.step = 0
+        episode = np.int16(0)
+        self.step = np.int16(0)
         observation = None
         episode_reward = None
         episode_step = None
@@ -124,8 +124,8 @@ def fit(self, env, nb_steps, action_repetition=1, callbacks=None, verbose=1,
             while self.step < nb_steps:
                 if observation is None:  # start of a new episode
                     callbacks.on_episode_begin(episode)
-                    episode_step = 0
-                    episode_reward = 0.
+                    episode_step = np.int16(0)
+                    episode_reward = np.float32(0)
 
                     # Obtain the initial observation by resetting the environment.
                     self.reset_states()
@@ -169,7 +169,7 @@ def fit(self, env, nb_steps, action_repetition=1, callbacks=None, verbose=1,
                 action = self.forward(observation)
                 if self.processor is not None:
                     action = self.processor.process_action(action)
-                reward = 0.
+                reward = np.float32(0)
                 accumulated_info = {}
                 done = False
                 for _ in range(action_repetition):