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"])