Skip to content

Commit

Permalink
[Doc] Update docstring for TruncatedNormal with correct parameter nam…
Browse files Browse the repository at this point in the history 
…es (#2625)

Co-authored-by: Valter Schutz <[email protected]>
  • Loading branch information
@valterschutz
valterschutz and Valter Schutz authored Dec 3, 2024
1 parent 8257799 commit d22266d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions torchrl/modules/distributions/continuous.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -205,8 +205,8 @@ class TruncatedNormal(D.Independent):
Default is 5.0
min (torch.Tensor or number, optional): minimum value of the distribution. Default = -1.0;
max (torch.Tensor or number, optional): maximum value of the distribution. Default = 1.0;
low (torch.Tensor or number, optional): minimum value of the distribution. Default = -1.0;
high (torch.Tensor or number, optional): maximum value of the distribution. Default = 1.0;
tanh_loc (bool, optional): if ``True``, the above formula is used for
the location scaling, otherwise the raw value is kept.
Default is ``False``;
Expand Down

1 comment on commit d22266d

@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 'CPU Benchmark Results'.
Benchmark result of this commit is worse than the previous benchmark result exceeding threshold 2.

Benchmark suite Current: d22266d Previous: 8257799 Ratio
benchmarks/test_replaybuffer_benchmark.py::test_rb_populate[TensorDictReplayBuffer-ListStorage-SamplerWithoutReplacement-400] 37.60959687575611 iter/sec (stddev: 0.1572274466797768) 227.8864254158803 iter/sec (stddev: 0.0008077828491354858) 6.06

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

CC: @vmoens

Please sign in to comment.