Setting requires_grad_ on gpu

[ptrblck]

Have a look at this post which explains this use case quite well.