您当前的位置: 首页 >  ui

柳鲲鹏

暂无认证

  • 0浏览

    0关注

    4642博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

问题解决:AttributeError: 'module' object has no attribute '_rebuild_tensor_v2'

柳鲲鹏 发布时间:2018-10-30 07:24:09 ,浏览量:0

 于训练模型时使用的是新版本的pytorch,而加载时使用的是旧版本的pytorch。

解决办法:

  • 升级pytorch

看起来理所当然,其实有可能有坑。说不定还要转回来。

  • 桥接

 在程序开头添加下面的代码,即可以使老版本pytorch兼容新版本pytorch,参考链接https://discuss.pytorch.org/t/question-about-rebuild-tensor-v2/14560

import torch._utils
try:
    torch._utils._rebuild_tensor_v2
except AttributeError:
    def _rebuild_tensor_v2(storage, storage_offset, size, stride, requires_grad, backward_hooks):
        tensor = torch._utils._rebuild_tensor(storage, storage_offset, size, stride)
        tensor.requires_grad = requires_grad
        tensor._backward_hooks = backward_hooks
        return tensor
    torch._utils._rebuild_tensor_v2 = _rebuild_tensor_v2
关注
打赏
1665724893
查看更多评论
立即登录/注册

微信扫码登录

0.0474s