Como verificar o tempo que meu modelo está na GPU em pytorch
next(network.parameters()).device
Difficult Duck
next(network.parameters()).device