如何训练具有多个输出的 pytorch 模型

Question

我正在尝试使用以下数据训练模型

图像是输入，必须预测 14 个特征。 请问我可以知道如何去训练这样的模型吗？ 谢谢。

Answer 1

就我而言，这些并不是真正的功能。这些是类，如果我没记错的话，您的图像有时属于多个类。

这是一个非常广泛的问题，但我认为here 可能是了解更多关于多标签图像分类的良好开端。

请注意，例如，您的模型不应与用于 cifar10 挑战的图像分类模型有太大不同。但是您需要构建数据并相应地选择损失函数。