Saving and loading Flux models containing optimizers

[rlrs]

When saving and loading a model as explained in the Flux docs, I get the error

ERROR: LoadError: MethodError: Cannot convert an object of type Vector{Pair{Any, Any}} to an object of type IdDict.

My model, however, is a bit more involved than a simple Flux Chain. The problem seems to be that it also includes an ADAM optimizer, which has a field called state with the value IdDict{Any,Any}(). I don't understand why this cannot be loaded?

[JonasIsensee]

Hi @rlrs,
can you give me a bit more information, please?
What JLD2 version are you on, which julia version?
Do you have any minimal code that produces the problem?

[rlrs]

Sorry for the lack of information. A pretty minimal example is this,

using Flux
using JLD2: @save, @load
opt = ADAM(1)
@save "test.jld2" opt
@load "test.jld2" opt

I'm on Julia 1.6.1, JLD2 v0.4.9, Flux v0.12.4.

[JonasIsensee]

Hi,
I implemented a fix in #330.
Does that resolve the problem for you as well?

[rlrs]

Thank you, it works in v0.4.10.