PublayNetデータセットで事前トレーニングされたカスタム データセットでDetectron2faster_rcnn_R_50_FPN_3xモデルをトレーニングしようとしています。トレーニング中に、次の警告が表示されます。
WARNING [01/14 14:35:22 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.cls_score.weight' to the model due to incompatible shapes: (7, 1024) in the checkpoint but (6, 1024) in the model! You might want to double check if this is expected.
WARNING [01/14 14:35:22 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.cls_score.bias' to the model due to incompatible shapes: (7,) in the checkpoint but (6,) in the model! You might want to double check if this is expected.
WARNING [01/14 14:35:22 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.bbox_pred.weight' to the model due to incompatible shapes: (24, 1024) in the checkpoint but (20, 1024) in the model! You might want to double check if this is expected.
WARNING [01/14 14:35:22 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.bbox_pred.bias' to the model due to incompatible shapes: (24,) in the checkpoint but (20,) in the model! You might want to double check if this is expected.
WARNING [01/14 14:35:22 fvcore.common.checkpoint]: Some model parameters or buffers are not found in the checkpoint:
roi_heads.box_predictor.bbox_pred.{bias, weight}
roi_heads.box_predictor.cls_score.{bias, weight}
さらに調査したところ、このリンクから、データセットに事前トレーニング済みモデルとは異なる数のクラスがあり、いくつかの事前トレーニング済み重みをスキップする必要があるため、上記の警告メッセージが予想されることがわかりました。
PublayNet には 5 つのクラスが{"text", "title", "list", "table", "figure"}あり、私のデータセットにはまさにこれら 5 つのクラスがあります。残念ながら、PublayNet で事前トレーニングされたモデルには 6 つのクラスがあります。5クラスじゃないの？この余分なクラスについて混乱しています。助けていただければ幸いです。