-
Notifications
You must be signed in to change notification settings - Fork 655
Open
Description
我训练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出问题了吗?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels