_pickle.UnpicklingError: invalid load key, '<'.
Micla-SHL opened this issue · comments
我训练ABCNet v2中文识别模型，可训练。在对保存的模型测试时候会出现这个问题
python tools/train_net.py --config-file /Micla/Project/AdelaiDet/output/batext/rects/v2_attn_R_50/config.yaml --num-gpus 1 --eval-only MODEL.WEIGHTS /Micla/Project/AdelaiDet/output/batext/rects/v2_attn_R_50/model_0029999.pth MODEL.BATEXT.EVAL_TYPE 3
我的训练环境是python3.8 pytorch 1.10.针对训练配置文件做了 NORM: "SyncBN" >修改成 "BN“ 因为我只有单GPU，batch_size 为1
其余没有更改，训练数据是原配置的ReCTS数据集。
我针对标题错误google提示是修改 self.CTLABELS = pickle.load(fp) ==> self.CTLABELS = pickle.loads(fp) 这没有起作用，会引入新错误 TypeError: a bytes-like object is required, not '_io.BufferedReader'
我知道的是pickle 在python内置，这是我python出问题了吗?
更新: 下载chn_cls_list 码表文件出问题, 虽然是提供了 - O 参数，但是我的梯子，跟终端没过多提示，我最开始也没有检查，采取未下载完成的chn_cls_list测试才导致如题问题，我的解决办法是将下载链接放到浏览器下载，浏览器能确保我的文件下载完成
wget https://drive.google.com/file/d/1XOtlUz9lxh2HV5Gmu3alb5WKZafFn-0_/view?usp=sharing -O model_v2_chn_pretrain.pth
wget https://drive.google.com/file/d/1wqkX2VAy48yte19q1Yn5IVjdMVpLzYVo/view?usp=sharing -O chn_cls_list