How to export trained agent in julia ReinforcementLearning

[Giotsik]

Hello, I would like to know how to export a trained agent in this package, in order to run it for several experiment scenarios

[findmyway]

In case you haven't seen this:

#183 (comment)

Because most models are constructed with Flux.jl, you can refer Saving and Loading Models in Flux.jl for more details.

[findmyway]

Cool. Glad it helps.

[Giotsik]

Hey sorry one more thing, in Pluto it seems to have this error, is it because of the gpu-cpu thing you mentioned? plus when I save it with cpu() it seems to show an error of not identifying the model

regards

[findmyway]

For the first question, you need to install Zygote first and using Zygote before loading parameters.
For the second question, it seems your code is invalid here.

Try ps = cpu(params(gdpolicy)) instead?

[Giotsik]

yeah sorry about the second. About the first I had already the package added and been using it :(

[findmyway]

That's strange. Could you check if your BSON is the latest version and try again in REPL instead of Pluto?