-
|
I really like tensordict, and I want to use it for most of module. Can I create a new module by inheriting TensorDictModule? Thanks! |
Beta Was this translation helpful? Give feedback.
Answered by
vmoens
Feb 7, 2025
Replies: 1 comment 1 reply
-
|
You should inherit from TensorDictModuleBase for anything with a TensorDict -> TensorDict signature! |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
Sorry for missing the email notification! I also read the code and understand what happens under the hood. elegant design! |
Beta Was this translation helpful? Give feedback.
Answer selected by
Roy-Kid
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
You should inherit from TensorDictModuleBase for anything with a TensorDict -> TensorDict signature!
(I should fix the docstrings of that thing btw!)