When I try to load the pretrained model. With command python predict.py --input datasets/data/cityscapes/leftImg8bit/train/bremen/bremen_000000_000019_leftImg8bit.png - -dataset cityscapes --model deeplabv3_resnet50 --ckpt ./deeplabv3/best_deeplabv3_resnet50_voc_os16.pth --save_val_results_to test_results, I got size mismatch for classifier.classifier.4.weight: copying a param with shape torch.Size([21, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([19, 256, 1, 1]). size mismatch for classifier.classifier.4.bias: copying a param with shape torch.Size([21]) from checkpoint, the shape in current model is torch.Size([19]).
When I try to load the pretrained model. With command
python predict.py --input datasets/data/cityscapes/leftImg8bit/train/bremen/bremen_000000_000019_leftImg8bit.png - -dataset cityscapes --model deeplabv3_resnet50 --ckpt ./deeplabv3/best_deeplabv3_resnet50_voc_os16.pth --save_val_results_to test_results, I gotsize mismatch for classifier.classifier.4.weight: copying a param with shape torch.Size([21, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([19, 256, 1, 1]). size mismatch for classifier.classifier.4.bias: copying a param with shape torch.Size([21]) from checkpoint, the shape in current model is torch.Size([19]).