Skip to content

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

@renshouyu

Description

@renshouyu

你好,我执行完当我执行完“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}

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions