判断model在cuda上

发表于 | 阅读次数: 
1
next(model.parameters()).is_cuda # returns a boolean
请作者喝一杯咖啡☕️