You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
In the next release, we'd like to incorporate the following new features:
## Objectives
Have all objectives work with native nn.Modules: We wish to be able to get all modules to work as in DQNLoss(nn.Linear(3, 4)) where the linear module is your value network. It's unclear how this will work when a distribution is needed (eg PPO). We will also have to overload the forward to patch it with the signature provided by @tensordict.dispatch and ensure compatibility with torch.compile. We also want to make this a blessed feature, presented clearly and concisely in the doc.
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
In the next release, we'd like to incorporate the following new features:
## Objectives
nn.Modules
: We wish to be able to get all modules to work as inDQNLoss(nn.Linear(3, 4))
where the linear module is your value network. It's unclear how this will work when a distribution is needed (eg PPO). We will also have to overload the forward to patch it with the signature provided by@tensordict.dispatch
and ensure compatibility with torch.compile. We also want to make this a blessed feature, presented clearly and concisely in the doc.Data
General
Beta Was this translation helpful? Give feedback.
All reactions