size mismatch for model.mapping

[CryoSky]

Hello again,

I tried to run the protein function annotation task after downloaded pkl file. However, I got the following error:

RuntimeError: Error(s) in loading state_dict for MultipleBinaryClassification: size mismatch for model.mapping: copying a param with shape torch.Size([20]) from checkpoint, the shape in current model is torch.Size([33]).

[sj4@gn10 ProtST]$ python ./script/run_downstream.py --config ./config/downstream_task/PretrainESM2/annotation_tune.yaml --checkpoint /work/sj4/protst_esm2.pth --dataset GeneOntology --branch BP

The yaml is file is almost the same as one on GitHub except I changed the number of GPUS to gpus: [0]. Could you please check this issue?

Best wishes,