model trick

文章目录

• 判断 model 在哪个设备上

判断 model 在哪个设备上

device = next(model.parameters()).device print(f"Model is on: {device}") 

is_cuda = next(model.parameters()).is_cuda if is_cuda:     print("Model is on GPU") else:     print("Model is on CPU")