From 309c0a3c615ab5f06b9aa2198f262ad04722605e Mon Sep 17 00:00:00 2001 From: Faury Louis Date: Thu, 13 Feb 2025 17:05:45 +0100 Subject: [PATCH] [Doc,BugFix] Fix default argument in doc for `AddStateIndependentNormalScale` (#1216) Co-authored-by: Louis Faury --- tensordict/nn/distributions/continuous.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensordict/nn/distributions/continuous.py b/tensordict/nn/distributions/continuous.py index 2c1c17c55..ff07f1369 100644 --- a/tensordict/nn/distributions/continuous.py +++ b/tensordict/nn/distributions/continuous.py @@ -98,7 +98,7 @@ class AddStateIndependentNormalScale(torch.nn.Module): Keyword Args: scale_mapping (str, optional): positive mapping function to be used with the std. - Defaults to ``"biased_softplus_1.0"`` (i.e. softplus map with bias such that fn(0.0) = 1.0) + Defaults to ``"exp"``, choices: ``"softplus"``, ``"exp"``, ``"relu"``, ``"biased_softplus_1"``. scale_lb (Number, optional): The minimum value that the variance can take. Defaults to ``1e-4``.