def forward(self, observations):
        """ Calculate model outputs """
        input_data = self.input_block(observations)

        base_output = self.backbone(input_data)
        policy_params = self.action_head(base_output)

        q = self.q_head(base_output)

        return policy_params, q