首页 > 解决方案 > 如何将 Pytorch 的 NLLLoss 与计数数据一起使用?

问题描述

我有一个模型,我试图用整数数据进行预测,其中预期输出的示例是 [40]。我如何使用 nn.PoissonNLLLoss 来做到这一点?设置 log_input=False 会慢慢将我的损失降低到 0 以下。

标签: modelintegerpytorch

解决方案


推荐阅读