def get_layer_groups(self):
return [<EOL>[self.policy_backbone, self.action_head],<EOL>[self.value_backbone, [y for (x, y) in self.critic_head.named_parameters() if x.endswith('<STR_LIT>')]],<EOL>[[y for (x, y) in self.critic_head.named_parameters() if x.endswith('<STR_LIT>')]]<EOL>]<EOL>
Return layers grouped