pytorch 读写t7

最新推荐文章于 2023-12-23 21:08:30 发布

AI算法网奇

最新推荐文章于 2023-12-23 21:08:30 发布

阅读量1w

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/84348225

版权

pytorch知识宝典专栏收录该内容

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了在使用PyTorch加载.t7文件时遇到的ASCII解码错误，提供了解决方案，通过设置pickle加载的encoding为'latin1'来避免报错。

摘要由CSDN通过智能技术生成

In [1]: import torchfile
In [2]: torchfile.load('testfiles_x86_64/list_table.t7')
Out[2]: ['hello', 'world', 'third item', 123]
In [3]: torchfile.load('testfiles_x86_64/doubletensor.t7')

上面代码可能报错：

'ascii' codec can't decode byte 0xc3

解决方法如下：

import pickle
pickle.load = partial(pickle.load, encoding="latin1")
pickle.Unpickler = partial(pickle.Unpickler, encoding="latin1")
self.net = torch.load('testfiles_x86_64/doubletensor.t7', map_location=lambda storage, loc: storage, pickle_module=pickle)