Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit 523cd9e

Browse files
author
Henryk Michalewski
committed
Reward changed to target_reward
1 parent f34a3a3 commit 523cd9e

File tree

1 file changed

+4
-4
lines changed
  • tensor2tensor/data_generators

1 file changed

+4
-4
lines changed

tensor2tensor/data_generators/gym.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -244,12 +244,12 @@ def hparams(self, defaults, unused_model_hparams):
244244
p = defaults
245245
p.input_modality = {"inputs": ("video", 256),
246246
"input_reward": ("symbol", self.num_rewards),
247-
"input_action": ("symbol", self.num_actions)}
247+
"input_action": ("symbol", self.num_actions)}
248248
# p.input_modality = {"inputs": ("video", 256),
249-
# "reward": ("symbol", self.num_rewards),
250-
# "input_action": ("symbol", self.num_actions)}
249+
# "reward": ("symbol", self.num_rewards),
250+
# "input_action": ("symbol", self.num_actions)}
251251
p.target_modality = {"targets": ("video", 256),
252-
"reward": ("symbol", self.num_rewards)}
252+
"target_reward": ("symbol", self.num_rewards)}
253253
#p.target_modality = {"targets": ("image", 256),
254254
# "reward": ("symbol", self.num_rewards + 1)} # ("video", 256)
255255
p.input_space_id = problem.SpaceID.IMAGE

0 commit comments

Comments
 (0)