Fix: clone of policy_outputs into logits in get_logprobs must not be …

…detached
alexhernandezgarcia · Nov 16, 2023 · 7ea8f9d · 7ea8f9d
1 parent 619281d
commit 7ea8f9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gflownet/envs/base.py b/gflownet/envs/base.py
@@ -546,7 +546,7 @@ def get_logprobs(
         """
         device = policy_outputs.device
         ns_range = torch.arange(policy_outputs.shape[0]).to(device)
-        logits = policy_outputs.clone().detach()
+        logits = policy_outputs.clone()
         if mask is not None:
             logits[mask] = -torch.inf
         action_indices = (