microsoft / RepPoints

Represent Visual Objects by Point Sets
MIT License
588 stars 75 forks source link

There's a problem when i test #23

Closed chen1234520 closed 4 years ago

chen1234520 commented 4 years ago

The model and loaded state dict do not match exactly

these keys have mismatched shape: +------------------------------------+-----------------------------+-----------------------------+ | key | expected shape | loaded shape | +------------------------------------+-----------------------------+-----------------------------+ | bbox_head.reppoints_cls_out.weight | torch.Size([80, 256, 1, 1]) | torch.Size([28, 256, 1, 1]) | | bbox_head.reppoints_cls_out.bias | torch.Size([80]) | torch.Size([28]) | +------------------------------------+-----------------------------+-----------------------------+

Thank you very much for help me!