def build_networks(self):
ac_net = ActorCriticNetworkDiscreteCNN(
list(self.env.observation_space.shape),
self.env.action_space.n,
self.config["n_hidden_units"],
summary=False)
return ac_net
def build_networks(self) -> ActorCriticNetworkDiscreteCNN:
return ActorCriticNetworkDiscreteCNN(
list(self.env.observation_space.shape),
self.env.action_space.n,
self.config["n_hidden_units"])