Skip to content

Commit

Permalink
[Doc,BugFix] Fix default argument in doc for `AddStateIndependentNorm…
Browse files Browse the repository at this point in the history 
…alScale` (#1216)

Co-authored-by: Louis Faury <[email protected]>
  • Loading branch information
@louisfaury
louisfaury and Louis Faury authored Feb 13, 2025
1 parent 0fccd26 commit 309c0a3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensordict/nn/distributions/continuous.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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``.
Expand Down

1 comment on commit 309c0a3

@github-actions
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Performance Alert ⚠️

Possible performance regression was detected for benchmark 'GPU Benchmark Results'.
Benchmark result of this commit is worse than the previous benchmark result exceeding threshold 2.

Benchmark suite Current: 309c0a3 Previous: 0fccd26 Ratio
benchmarks/tensorclass/test_torch_functions.py::test_ones_like 113.93788946510738 iter/sec (stddev: 0.0028169078278795986) 241.43491416269896 iter/sec (stddev: 0.00010233623374763178) 2.12

This comment was automatically generated by workflow using github-action-benchmark.

CC: @vmoens

Please sign in to comment.