Fix sac.entropy loss assert.

learnables · Sep 9, 2019 · cce50a6 · cce50a6
1 parent 6aca5a0
commit cce50a6
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/cherry/algorithms/sac.py b/cherry/algorithms/sac.py
@@ -220,8 +220,6 @@ def entropy_weight_loss(log_alpha, log_probs, target_entropy):
     ~~~
 
     """
-    msg = 'v_value, q_value, and log_probs must have equal size.'
-    assert log_alpha.size() == log_probs.size(), msg
     if debug.IS_DEBUGGING:
         if log_probs.requires_grad:
             debug.logger.warning('SAC:entropy_weight_loss: log_probs.requires_grad is True.')