我之前在 Flux.jl 中训练了一个模型并通过以下操作保存了它:
@save "mymodel.bson" model
现在我想重新加载该模型并再次使用它。如何在 Flux 中实现这一点?
最佳答案
与上面使用的 @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)
您可以了解更多关于 saving and loading models in the Flux.jl docs 的信息.
关于julia - 如何在 Flux.jl 中使用 BSON 加载经过训练的模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/68335891/