Method features

4-atari-hard/3-robustify.py:94–96

(self, obs)

        self.gru_dim = gru_dim

    def features(self, obs):
        h = self.ln(torch.relu(self.fc(self.conv(obs / 255.0))))
        return h

    def step(self, obs, hx, inc_entropy=None):
        """One timestep for acting. obs (B,4,105,80), hx (B,gru). Returns