gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00
Code Issues Projects Releases Wiki Activity

add meaningful error message in the event that the action space is not one that can be used (#151)

Browse Source
This commit is contained in:
Zach Dwiel
2018-12-11 02:09:24 -05:00
committed by Gal Leibovich
parent f12857a8c7
commit d0248e03c6
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rl_coach/architectures/tensorflow_components/heads/q_head.py
View File

@@ -34,6 +34,12 @@ class QHead(Head):
self.num_actions = 1 self.num_actions = 1
elif isinstance(self.spaces.action, DiscreteActionSpace): elif isinstance(self.spaces.action, DiscreteActionSpace):
self.num_actions = len(self.spaces.action.actions) self.num_actions = len(self.spaces.action.actions)
else:
raise ValueError(
'QHead does not support action spaces of type: {class_name}'.format(
class_name=self.spaces.action.__class__.__name__,
)
)
self.return_type = QActionStateValue self.return_type = QActionStateValue
if agent_parameters.network_wrappers[self.network_name].replace_mse_with_huber_loss: if agent_parameters.network_wrappers[self.network_name].replace_mse_with_huber_loss:
self.loss_type = tf.losses.huber_loss self.loss_type = tf.losses.huber_loss
@@ -49,5 +55,3 @@ class QHead(Head):
"Dense (num outputs = {})".format(self.num_actions) "Dense (num outputs = {})".format(self.num_actions)
] ]
return '\n'.join(result) return '\n'.join(result)