RuntimeError: mat1 dim 1 must match mat2 dim 0

[waleed-27]

Hi, I'm trying to run your code using the following command

python run_sglayer.py --neibor relation -e res50 --gpu 0 --batch-size 4 --train-percent 0.7 -r all --pps totally >./gcn_CXR_res50_aps.log

And I am getting the following error:

Can you tell me how to fix it? Thank you