python - 我们如何根据条件创建具有多个输出的 pytorch 模型？

Question

我想制作一个多输出模型，其中有两种不同的条件。模型应该是这样的

def forward:
    if condition1 and condition2:
         something_1
    if condition1:
         something_2
    if condition2:
         something_3

    o1 = classifier_output(something_1)
    o2 = classifier_output(something_2)
    o3 = classifier_output(something_3)

    return{o1, o2, o3}

我在这里遇到的主要问题是，当我们只有条件 1 或条件 2 为真时，有时不会调用其中一个 if 条件。在那种情况下，如何提取输出分类器的 logits？根据获得的输出数量，返回是否也会成为条件语句的一部分？

Answer 1

您的第一个条件是and，因此如果您的条件 1 或 2 为 False，则只会运行一个 if 语句。在这种情况下，您需要为 something_x 设置一些默认值，否则它会发生冲突，或者您可以将 o1,o2,o3 放入 if 语句中，并为 o1,o2,o3 上方的 o1,o2,o3 提供一些初始值if 语句。