Skip to content

[Feature Request] All models in modules.models should have a device kwarg #507

Closed
@vmoens

Description

@vmoens

Motivation

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.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions