f056fb8f6226c83d340c869e0d5312d61acf07f0,src/garage/torch/q_functions/discrete_dueling_cnn_q_function.py,DiscreteDuelingCNNQFunction,forward,#DiscreteDuelingCNNQFunction#Any#,115

Before Change

            obs_shape = ((len(observations), ) +
                         self._env_spec.observation_space.shape)
            observations = observations.reshape(obs_shape)
        return super().forward(observations)

After Change

            torch.Tensor: Output value
        
        // We"re given flattened observations.
        observations = observations.reshape(
            -1, *self._env_spec.observation_space.shape)
        out = self._module(observations)
        val = self._val(out)
        act = self._act(out)
        act = act - act.mean(1).unsqueeze(1)
        return val + act

Instances