pytorch版本問題:AttributeError: 'module' object has no attribute '_rebuild_tensor_v2'


用pytorch加載訓練好的模型的時候遇到了如下的問題:

AttributeError: 'module' object has no attribute '_rebuild_tensor_v2'

 

到網上查了一下是由於訓練模型時使用的是新版本的pytorch,而加載時使用的是舊版本的pytorch。

解決方法:

1、既然是pytorch版本較老,那最簡單的解決方法當然是簡單的升級一下pytorch就ok了。

2、國外的大神給了另一種解決方法,就是在程序開頭添加下面的代碼,即可以使老版本pytorch兼容新版本pytorch,參考鏈接https://discuss.pytorch.org/t/question-about-rebuild-tensor-v2/14560

 1 import torch._utils
 2 try:
 3     torch._utils._rebuild_tensor_v2
 4 except AttributeError:
 5     def _rebuild_tensor_v2(storage, storage_offset, size, stride, requires_grad, backward_hooks):
 6         tensor = torch._utils._rebuild_tensor(storage, storage_offset, size, stride)
 7         tensor.requires_grad = requires_grad
 8         tensor._backward_hooks = backward_hooks
 9         return tensor
10     torch._utils._rebuild_tensor_v2 = _rebuild_tensor_v2

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM