gpt4 book ai didi

pytorch - 在pytorch中,损失函数需要继承nn.module的情况有哪些?

转载 作者:行者123 更新时间:2023-12-04 09:49:16 24 4
gpt4 key购买 nike

我对 PyTorch 中的损失函数感到困惑。有些人将损失函数定义为普通的python函数,而另一些人通过定义一个继承nn.Module的类来定义损失函数。所以想知道什么情况下我们需要通过继承nn.Module来定义损失函数?非常感谢。

最佳答案

一般来说,继承自 nn.Module只有当你想在这个模块中有可训练的变量时才需要,否则继承它是可选的。

所以同样适用于损失函数,如果它不包含这样的变量(我认为这是主要情况),则不需要继承。

关于pytorch - 在pytorch中,损失函数需要继承nn.module的情况有哪些?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62038560/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com