Model forward pass

def forward(self, observations):
        """ Model forward pass """
        observations = self.input_block(observations)
        advantage_features, value_features = self.backbone(observations)
        q_values = self.q_head(advantage_features, value_features)

        return q_values