f METHOD_NAME(self,
              networks: value_dice_networks.ValueDiceNetworks,
              environment_spec: specs.EnvironmentSpec,
              evaluation: bool = False) -> actor_core_lib.FeedForwardPolicy: