|
|
|
|
|
|
:param brain: State-space size |
|
|
|
:param h_size: Hidden layer size |
|
|
|
""" |
|
|
|
super().__init__() |
|
|
|
super(ContinuousControlModel, self).__init__() |
|
|
|
s_size = brain.state_space_size |
|
|
|
a_size = brain.action_space_size |
|
|
|
|
|
|
|
|
|
|
:param brain: State-space size |
|
|
|
:param h_size: Hidden layer size |
|
|
|
""" |
|
|
|
super().__init__() |
|
|
|
super(DiscreteControlModel, self).__init__() |
|
|
|
self.create_global_steps() |
|
|
|
self.create_reward_encoder() |
|
|
|
self.normalize = normalize |
|
|
|