php网站留言板怎么做,wordpress 相关文章推荐,有单做平台app,世界500强设计公司有哪些1、 这句话的意思是#xff1a;期望张量的后端处理是在cpu上#xff0c;但是实际是在cuda上。排查代码发现#xff0c;数据还在cpu上#xff0c;但是模型已经转到cuda上#xff0c;所以可以通过把数据转到cuda上解决。 解决代码#xff1a;
tensor.to(cuda)…1、 这句话的意思是期望张量的后端处理是在cpu上但是实际是在cuda上。排查代码发现数据还在cpu上但是模型已经转到cuda上所以可以通过把数据转到cuda上解决。 解决代码
tensor.to(cuda)2、 解决方法减小batch size的大小或减小图片的尺寸 3、
原始的代码如下
torch.save(model.module.state_dict(), os.path.join(model_prefix, {}-model.pth.format(epoch)))原因因为只有一块GPU以上代码是多GPU使用的 解决方法改为以下单GPU代码 meta {}checkpoint {meta: meta, state_dict: weights_to_cpu(model.state_dict())}if optimizer is not None:checkpoint[optimizer] optimizer.state_dict()torch.save(checkpoint, os.path.join(model_prefix, {}-model.pth.format(epoch)))
