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
We can now create most nn.Module on device directly by passing a device kwarg. However this is not the case in TorchRL's modules. Classes such as MLP and ConvNet should have a device kwarg that is passed to all layers.