标签: pytorch cnn facial-identification
我正在尝试使用PyTorch创建人脸识别模型。我的模型在训练和验证的损失得分都接近于0的情况下表现良好。问题是,当我使用相同的输入对其进行测试时,它给我的结果如下所示:
及以下
左侧是真实标签,右侧是预测输出。我设置随机种子并运行model.eval(),然后再将输入提供给模型。
model.eval()
有人知道如何解决吗?
请告知