当我阅读 Flux 文档时，我看到为我们定义了许多不同的损失函数，我们可以使用它们。我知道损失告诉我们距离目标值还有多远。但是我在训练循环中实际在哪里使用损失函数呢？
1 回答
1
如果您使用的是内置train!()函数，您可以定义损失函数并在训练期间使用它，如下所示：
loss(x, y) = Flux.Losses.mse(m(x), y)
ps = Flux.params(m)
Flux.train!(loss, ps, data, opt)
其中使用内置的均方误差函数来计算和Flux.Losses.mse之间的距离。您可以在此处阅读有关 Flux 中损失函数的更多信息：https ://fluxml.ai/Flux.jl/stable/training/training/#Loss-Functionsm(x)y
于 2021-07-04T23:06:06.300 回答