首页 > 解决方案 > 将图外计算的损失分配给 Pytorch 模型

问题描述

我需要找到一种在 pytorch 模型上使用非 pytorch 损失函数的方法。除了重写 pytorch 损失系统之外,有没有办法将计算出的损失分配给模型图,以便可以与.backwards? 创建人工计算图是否可行?

标签: neural-networkpytorchloss-function

解决方案


推荐阅读