def forward(self, observations):
input_data = self.input_block(observations)<EOL>policy_base_output = self.policy_backbone(input_data)<EOL>value_base_output = self.value_backbone(input_data)<EOL>action_output = self.action_head(policy_base_output)<EOL>value_output = self.value_head(value_base_output)<EOL>return action_output, value_output<EOL>
Calculate model outputs