【发布时间】:2021-07-04 23:06:06
【问题描述】:
当我阅读 Flux 文档时,我看到有一个为我们定义的 bunch of different loss functions 可以使用。我知道损失告诉我们距离目标值还有多远。但是我在训练循环中实际在哪里使用损失函数呢?
【问题讨论】:
当我阅读 Flux 文档时,我看到有一个为我们定义的 bunch of different loss functions 可以使用。我知道损失告诉我们距离目标值还有多远。但是我在训练循环中实际在哪里使用损失函数呢?
【问题讨论】:
如果您使用的是内置的train!() 函数,您可以定义损失函数并在训练期间使用它,如下所示:
loss(x, y) = Flux.Losses.mse(m(x), y)
ps = Flux.params(m)
Flux.train!(loss, ps, data, opt)
其中Flux.Losses.mse 使用内置的均方误差函数来计算m(x) 和y 之间的距离。您可以在此处阅读有关 Flux 中损失函数的更多信息:https://fluxml.ai/Flux.jl/stable/training/training/#Loss-Functions
【讨论】: