如何在 Flux.jl 中使用 BSON 加载经过训练的模型

Question

我之前在 Flux.jl 中训练了一个模型并通过执行以下操作来保存它：

@save "mymodel.bson" model

现在我想重新加载该模型并再次使用它。我怎样才能在 Flux 中实现这一点？

Answer 1

与@save上面使用的宏类似，还有一个@load来自 BSON 包的内置宏。您可以通过执行以下操作来访问它using BSON: @load，然后非常简单地执行以下操作：

julia> using Flux

julia> using BSON: @load

julia> @load "mymodel.bson" model

julia> model
Chain(Dense(10, 5, NNlib.relu), Dense(5, 2), NNlib.softmax)

您可以在 Flux.jl 文档中找到有关保存和加载模型的更多信息。