最近要把caffe模型参数来初始化pytorch模型,想法是先把caffe模型转成dict,再跟pytorch的参数匹配。在这个过程中需要用到pickle.load()函数,但在使用时却遇到了以下编码错误:UnicodeDecodeError: 'ascii' codec can't decode byte 0x8b in position 6: ordinal not in range(128)。把pickle.load()函数中的encoding设为'iso-8859-1'便可解决问题,
import pickle as pkl
weights = "xxx.pkl"
pkl.load(weights, encoding="iso-8859-1")