如何在具有不同类的数据集上微调模型?

2024-09-30 06:30:47 发布

您现在位置:Python中文网/ 问答频道 /正文

在本paper中,作者对COCO预训练模型进行了微调,使其适合PoseTrack数据集。我想做一些类似的事情,但我不知道怎么做,因为类(键联接)的数量不一样

我曾想过“禁用”某些类,但我不确定如何禁用


Tags: 数据模型数量作者事情papercoco微调
1条回答
网友
1楼 · 发布于 2024-09-30 06:30:47

您需要根据问题的类数更改nn的最后一层,并为数据集重新训练(假设它具有相同的输入形状)。您可以使用前一层的预训练权重,以便收敛速度更快,所需数据更少

相关问题 更多 >

    热门问题