How to check if Model is on cuda

[smth]

As replied on the github issues, an easy way is:

next(model.parameters()).is_cuda # returns a boolean