Merge pull request #251 from alexhernandezgarcia/fix-get_logprobs-base

Fix: get_logprobs of base env (reason why TB was not training well)
alexhernandezgarcia · Nov 17, 2023 · cb4bf18 · cb4bf18
2 parents 619281d + 7ea8f9d
commit cb4bf18
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 = (