仅使用 RESUME 参数时 mAP 很高,否则几乎为零

Hi @alcinos、@fmassa， 我正在使用自定义数据集训练 DETR。我在代码中做了以下更改 hubconf.py: if pretrained: checkpoint = torch.load("detr-r50-e632da11.pth",map_location="cpu") del checkpoint["model"]["class_embed.weight"] del checkpoint["model"]["class_embed.bias"] torch.save(checkpoint,"detr-r50_no-class-head.pth") main.py: model_without_ddp.load_state_dict(checkpoint['model'],strict=False)

我训练了 500 次迭代，但 mAP 仍然只有零左右。然而，当我使用 "detr-r50_no-class-head.pth" 重新开始训练时，mAP 突然飙升到 0.72。为了再次检查，我使用 21 次迭代训练了模型，然后使用 RESUME 参数重新开始训练。它再次显示 mAP 为 0.72。请参阅下面的图表