fixes to saving
This commit is contained in:
parent
6fd66fd3b5
commit
b08c949b99
|
@ -53,7 +53,7 @@ julia> using Flux
|
|||
julia> model = Chain(Dense(10,5,relu),Dense(5,2),softmax)
|
||||
Chain(Dense(10, 5, NNlib.relu), Dense(5, 2), NNlib.softmax)
|
||||
|
||||
julia> weights = Tracker.data.(params(model));
|
||||
julia> weights = params(model);
|
||||
|
||||
julia> using BSON: @save
|
||||
|
||||
|
|
Loading…
Reference in New Issue