COCO pretrained model dimension

[tghim]

Thank you so much for sharing your great work. When I tested this network to coco dataset with your coco pretrained model, The dimension issue happens.

"size mismatch for decoder.last_conv.8.bias: copying a param with shape torch.Size([17]) from checkpoint, the shape in current model is torch.Size([18])."

An error message appears above, the decoder adds 1 to numclasses, so the dimensions don't match with the pretrained model.

I would be very grateful if you could reply.