摘要: 最近遇到一个问题,发下gpu训练的模型在cpu下不能跑,很难受。结果发现原来是自己使用了DataParallel的问题,我保存的是整个模型。 model=DataParalle(model)torch.save(‘xx.pkl’,model) 然后把模型copy到自己电脑上用cpu跑 model=t 阅读全文
posted @ 2022-03-17 09:34 笨笨和呆呆 阅读(1247) 评论(0) 推荐(0)