From 1bab6a9d1fce390cd08d973706a143c1d315618e Mon Sep 17 00:00:00 2001 From: Timo <43245438+TheRisenPhoenix@users.noreply.github.com> Date: Thu, 2 May 2024 14:39:54 +0200 Subject: [PATCH] Fix doc default argument mismatch --- torchrl/objectives/sac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrl/objectives/sac.py b/torchrl/objectives/sac.py index 0c7defbe853..d30efb4e1a1 100644 --- 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.