时间戳

首页 > 解决方案 > 如何在 Flux.jl 中使用损失函数

julia - 如何在 Flux.jl 中使用损失函数

问题描述

当我阅读 Flux 文档时，我看到为我们定义了许多不同的损失函数，我们可以使用它们。我知道损失告诉我们距离目标值还有多远。但是我在训练循环中实际在哪里使用损失函数呢？

标签： juliaflux.jl

解决方案

如果您使用的是内置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

推荐阅读