[Doc] Fix documentation mismatch for default argument (#2149)

pytorch · May 2, 2024 · 48b5a28 · 48b5a28
1 parent 7109a3f
commit 48b5a28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/objectives/sac.py b/torchrl/objectives/sac.py
@@ -802,7 +802,7 @@ class DiscreteSACLoss(LossModule):
             :class:`torchrl.data.BinaryDiscreteTensorSpec` or :class:`torchrl.data.DiscreteTensorSpec`).
         num_actions (int, optional): number of actions in the action space.
             To be provided if target_entropy is set to "auto".
-        num_qvalue_nets (int, optional): Number of Q-value networks to be trained. Default is 10.
+        num_qvalue_nets (int, optional): Number of Q-value networks to be trained. Default is 2.
         loss_function (str, optional): loss function to be used for the Q-value. Can be one of `"smooth_l1"`, "l2",
             "l1", Default is "smooth_l1".
         alpha_init (float, optional): initial entropy multiplier.