【问题标题】:How to use a Loss function in Flux.jl如何在 Flux.jl 中使用损失函数
【发布时间】:2021-07-04 23:06:06
【问题描述】:

当我阅读 Flux 文档时,我看到有一个为我们定义的 bunch of different loss functions 可以使用。我知道损失告诉我们距离目标值还有多远。但是我在训练循环中实际在哪里使用损失函数呢?

【问题讨论】:

    标签: julia flux.jl


    【解决方案1】:

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2021-09-20
      • 2020-10-22
      • 2021-11-06
      • 1970-01-01
      • 1970-01-01
      • 2017-10-16
      相关资源
      最近更新 更多