ucbdrive / few-shot-object-detection

Implementations of few-shot object detection benchmarks
Apache License 2.0
1.1k stars 223 forks source link

checkpoints文件夹下没有生成model_final.pth文件 #198

Open renshouyu opened 7 months ago

renshouyu commented 7 months ago

你好,我执行完当我执行完“python3 -m tools.train_net --num-gpus 1 --config-file configs/PascalVOC-detection/split1/,完成基类训练后,checkpoints下没有生成model_final.pth文件,在inference下有个res_final.json,训练时有一些warning, WARNING [03/06 20:16:49 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.cls_score.weight' to the model due to incompatible shapes: (16, 1024) in the checkpoint but (5, 1024) in the model! You might want to double check if this is expected. WARNING [03/06 20:16:49 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.cls_score.bias' to the model due to incompatible shapes: (16,) in the checkpoint but (5,) in the model! You might want to double check if this is expected. WARNING [03/06 20:16:49 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.bbox_pred.weight' to the model due to incompatible shapes: (60, 1024) in the checkpoint but (16, 1024) in the model! You might want to double check if this is expected. WARNING [03/06 20:16:49 fvcore.common.checkpoint]: Skip loading parameter 'roi_heads.box_predictor.bbox_pred.bias' to the model due to incompatible shapes: (60,) in the checkpoint but (16,) in the model! You might want to double check if this is expected. WARNING [03/06 20:16:49 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}