gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-04 16:24:24 27 4
gpt4 key购买 nike

我之前在 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/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com