使用pickle时出现UnicodeDecodeError的解决方法

最近要把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")