预训练Pytorch模型跨python2/3版本读入的编码问题

在尝试跨Python版本读取预训练的Pytorch模型时,遇到编码问题,官方建议的解决方案无法直接解决。该问题源于Python 2/3对于str编码方式的差异。一种有效的方法是先在Python2环境下读取并重新保存模型的'state_dict',然后在Python3环境下重新加载。按照step 1、step 2和step 3进行操作可以成功解决这个问题。
摘要由CSDN通过智能技术生成

当想读入现有的模型,而且并不是同意台电脑预训练的模型,而且需要跨python版本,就会出现下面的情况。
img
官方也有issue说这个, 里面的方法我并没有解决问题
在这里插入图片描述
上面的方法读出来只是一个int,至于为什么还没弄清。

虽然官方说已经在写好patch,但是大概没有在现有版本中release. github.request


这其实是python 2/3 对于str的编码方式的差别的问题。

源码中的注释有写到:

.. note::
        In Python 3, when loading files saved by Python 2, you may encounter
        ``UnicodeDecodeError: 'ascii' codec can't decode byte 0x...``. This is
        caused by the difference of handling in byte strings in Python2 and
        Python 3. You may use extra ``encoding`` keyword argument t
  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值