Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Extract primers from modules that contain RNNs #2127

Merged
merged 11 commits into from
May 3, 2024
Merged

[Feature] Extract primers from modules that contain RNNs #2127

Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
221756f
get_primers_from_module
albertbou92 Apr 28, 2024
3cc8ea8
format
albertbou92 Apr 28, 2024
918a1b9
format
albertbou92 Apr 28, 2024
320d873
format
albertbou92 Apr 29, 2024
96ad080
Merge remote-tracking branch 'origin/main' into extract_module_primers
vmoens Apr 30, 2024
35d7e56
fix import and header
albertbou92 Apr 30, 2024
240d5bd
add to doc
albertbou92 Apr 30, 2024
232dbb4
add example
albertbou92 Apr 30, 2024
97696d3
add cross refs
albertbou92 Apr 30, 2024
0937e79
Apply suggestions from code review
vmoens Apr 30, 2024
f48eb0c
add test
albertbou92 Apr 30, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
add to doc
  • Loading branch information
@albertbou92
albertbou92 committed Apr 30, 2024
commit 240d5bd234823e1f90b635b2dc7e6aa945b18e91
1 change: 1 addition & 0 deletions docs/source/reference/modules.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -427,6 +427,7 @@ Utils
mappings
inv_softplus
biased_softplus
get_primers_from_module

.. currentmodule:: torchrl.modules

Expand Down