policy_loss in SAC_CQL is significantly higher than the official implementation when tested with hopper-expert-v0 in d4rl.
|
policy_loss = ((self.alpha * log_pi) - min_qf_pi).mean() |
With the author's implementation, we can get the loss lower than -350, while using accel we can't even reach -300, which leads to slower and unstable learning.
policy_lossin SAC_CQL is significantly higher than the official implementation when tested withhopper-expert-v0in d4rl.accel/accel/agents/sac_cql.py
Line 261 in af3f511
With the author's implementation, we can get the loss lower than -350, while using accel we can't even reach -300, which leads to slower and unstable learning.