'roi_heads.box_predictor.cls_score.weight' has shape (81, 1024) in the checkpoint but (157, 1024) in the model! Skipped.

boingboing·2024년 3월 5일
  • The warning message is expected. Because your dataset has different number of classes from the pre-trained model, a few pre-trained weights have to be skipped.
  • pre-trained 모델의 클래스 갯수와 , 데이터셋의 클래스 갯수가 다름

-> 이 warning이 뜨긴 했는데 Bounding Box Regression과 Mask Loss가 0이 된 원인은 아닌 듯.

0개의 댓글