We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 60d8b1a commit f61078dCopy full SHA for f61078d
torchrl/objectives/ppo.py
@@ -677,7 +677,7 @@ def loss_critic(self, tensordict: TensorDictBase) -> torch.Tensor:
677
loss_value, clip_fraction = _clip_value_loss(
678
old_state_value,
679
state_value,
680
- self.clip_value.to(state_value.device),
+ self.clip_value,
681
target_return,
682
loss_value,
683
self.loss_critic_type,
0 commit comments