spanev:

What is device_ here?


It is;
from torch import device as device_

And the Model does not have argument of the device.

I mean is when having the device as an argument, how using it in the init.