wlyzyz的博客

原创 二分类问题 模型输出层+Loss 探究

二分类任务中，一般来说只有3种写法：1. 模型输出层：Linear(hidden_size, 2)，Loss：nn.CrossEntropyLoss()2. 模型输出层：Linear(hidden_size, 1)，Loss：nn.BCEWithLogitsLoss()3. 模型输出层：Sigmoid(Linear(hidden_size, 1))，Loss：nn.BCELoss()但最近我发现还有第四种写法：4. 模型输出层：Sigmoid(Linear(hidden_size, 1))，Loss：nn.