Skip to content

Commit a5efb4e

Browse files
Removed entropy term (#1)
1 parent c5bc5ce commit a5efb4e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stable_baselines3/rsppo/rsppo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@ def train(self) -> None:
255255

256256
entropy_losses.append(entropy_loss.item())
257257

258-
loss = policy_loss + self.ent_coef * entropy_loss + self.vf_coef * value_loss
258+
loss = policy_loss + self.vf_coef * value_loss
259259

260260
# Calculate approximate form of reverse KL Divergence for early stopping
261261
# see issue #417: https://github.com/DLR-RM/stable-baselines3/issues/417

0 commit comments

Comments
 (0)