def forward(self, observations):
input_data = self.input_block(observations)<EOL>base_output = self.backbone(input_data)<EOL>policy_params = self.action_head(base_output)<EOL>q = self.q_head(base_output)<EOL>return policy_params, q<EOL>
Calculate model outputs