python - 使用 2 组和/或 2 个差异损失函数进行正确训练是否正常？

Question

我有 2 套火车：一套有标签，一套没有标签。

训练时，我同时从标记集中加载一个批次，然后使用第一个损失函数进行计算；然后，来自未标记集合的一批，使用另一个函数计算。最后，我将它们相加（2 次损失）和loss.backward().

这种方式行得通吗？在我看来这很罕见，所以只要问一下引擎是否知道如何正确地反向传播（没有错）？谢谢你。

标签： pythonpytorchloss-function

Answer 1

我从 pytorch 论坛讨论中得到了答案。pytorch 中的 Autograd 引擎使用 2 个或更多不同的损失函数可以正常工作，因此我们不必担心它的准确性。谢谢