How to check if Model is on cuda

9

You can get the device by:

next(network.parameters()).device
15 Likes