torch.load取模型的cpu设置，仅一行代码（当cuda不能使用时，或cuda out of memory时，可以用cpu跑这个gpu训练的模型）

仅作为记录，大佬请跳过。

用map_location='cpu'

model = torch.load('model25_152_finetune_invade_bestmodel_v2_v2_Untitled3.pkl',map_location='cpu')
版权声明：本文为weixin_41529093原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。