I print the weight it shows following.
[ 0.0092,  0.1345, -0.0752]]]], requires_grad=True).
No cuda:0 but when i print(model.device) before that it showing cuda:0.
If we assign GPU to model, can’t layer weight will also assign
the GPU. where is the conflict?