In the current implementation of compute_policy_loss_clip_cov, I noticed that log_prob is not detached from the computation graph during the covariance matrix calculation. While this currently does ...