2

我正在尝试使用三元组损失函数对 ResNet50V2 模型进行迁移学习。我保留了 Include_top = False, input shape = (160,160,3) 和 Imagenet 权重。我的模型的最后 3 层如下图所示,有 600 万个可训练参数。

在此处输入图像描述

在训练过程中,我可以看到损失函数值从 7.6 降低到 0.8,但准确度没有提高。但是当我用 VGG16 替换模型并训练最后 3 层时,准确率从 50% 提高到 90%,同时损失值从 6.0 降低到 0.5。

我哪里错了?在训练 resnet 模型时我应该看什么具体的东西吗?如何训练 resnet 模型?

4

0 回答 0