Which device is model / tensor stored on?

14

Is there a way to get the device of a module (a torchvision model or criterion) ?

1 Like