标签:with tee class ror tps 内容 error: storage when
- pytorch0.4有一些接口已经改变,且模型向下版本兼容,不向上兼容。
使用pytorch0.3导入pytorch0.4保存的模型时候:
Monkey-patch because I trained with a newer version. # This can be removed once PyTorch 0.4.x is out. # See 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
标签:with tee class ror tps 内容 error: storage when
原文地址:https://www.cnblogs.com/ranjiewen/p/9484323.html