Как загрузить обученную модель с BSON в Flux.jl

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 документации .