def METHOD_NAME(self, metrics: dict, experience: Experience) -> None:
    if self.disable:
        return
    self.learn_timer.end()
    batch_size, seq_len = experience.sequences.shape
    self.learn_num_samples += batch_size
    # actor forward-backward, 3 means forward(1) + backward(2)
    self.learn_flop += self.actor_num_params * batch_size * seq_len * 2 * (3 + int(self.enable_grad_checkpoint))
    # critic forward-backward
    self.learn_flop += self.critic_num_params * batch_size * seq_len * 2 * (3 + int(self.enable_grad_checkpoint))