Pytorch Get Device Of Model