如何在 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 文檔中找到有關保存和加載模型的更多信息。