I don’t know, if there is a proper method for built-in nn.Modules, but if you are concerned about profiling or debugging, you could create a custom module and create the parameters directly on the device.