test.py error

[yong6600]

I have trouble when test.py command I want solve that this problem

This is test.py command

python test.py --model_path ./runs/fcn16s_cityscapes_best_model.pkl --img_path ~/dataset/cityscape/leftImg8bit/test/berlin/berlin_000197_000019_leftImg8bit.png --out_path ./out/

error

File "test.py", line 157, in test(args) File "test.py", line 60, in test model.load_state_dict(state) File "/home/yong/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/nn/modules/module.py", line 777, in load_state_dict self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for fcn16s: size mismatch for classifier.6.weight: copying a param with shape torch.Size([19, 4096, 1, 1]) from checkpoint, the shape in current model is torch.Size([21, 4096, 1, 1]). size mismatch for classifier.6.bias: copying a param with shape torch.Size([19]) from checkpoint, the shape in current model is torch.Size([21]). size mismatch for score_pool4.weight: copying a param with shape torch.Size([19, 512, 1, 1]) from checkpoint, the shape in current model is torch.Size([21, 512, 1, 1]). size mismatch for score_pool4.bias: copying a param with shape torch.Size([19]) from checkpoint, the shape in current model is torch.Size([21]).

[callmediddy]

Did you figure out the answer to your question? I ran into a similar error.

[givenjiang]

21 means the n_classes parameter in test.py, you can change it from 21 to 19 manually to fix the error.