如何在 Flux.jl 中指定我的模型应该训练的纪元数

Question

我知道我想根据我在其他框架和以前的模型中完成的一些模拟测试，让我的模型训练 200 个 epoch。我如何在 Flux.jl 中指定我的模型应该训练的 epoch 数？

Answer 1

Flux 提供了一个名为 @epochs 的便利宏，可以如下使用：

julia> using Flux: @epochs

julia> Flux.@epochs 2 println("hello")
[ Info: Epoch 1
hello
[ Info: Epoch 2
hello

或者就我们的训练循环而言：

julia> using Flux: @epochs

julia> @epochs 2 Flux.train!(...)
# Train for two epochs

您可以在此处从 Flux.jl 文档中找到更多信息：https://fluxml.ai/Flux.jl/stable/training/training/