如何在 Flux.jl 中使用 BSON 加载经过训练的模型
How to load a trained model with BSON in Flux.jl
我在 Flux.jl 之前训练了一个模型并通过以下操作保存了它：
@save "mymodel.bson" model
现在我想重新加载该模型并再次使用它。我怎样才能在 Flux 中实现这一点？
类似于上面使用的@save宏，还有一个来自BSON包的内置@load宏。您可以通过执行 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)
您可以了解有关 saving and loading models in the Flux.jl docs 的更多信息。
我在 Flux.jl 之前训练了一个模型并通过以下操作保存了它：
@save "mymodel.bson" model
现在我想重新加载该模型并再次使用它。我怎样才能在 Flux 中实现这一点？
类似于上面使用的@save宏，还有一个来自BSON包的内置@load宏。您可以通过执行 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)
您可以了解有关 saving and loading models in the Flux.jl docs 的更多信息。