How to save trained network weights?

flow-project / flow

Computational framework for reinforcement learning in traffic control

MIT License

1.04k stars 369 forks source link

How to save trained network weights? #1046

Open Zhenlin-Xu opened 2 years ago

Zhenlin-Xu commented 2 years ago

I have already trained a agent and I would like to use the weights of nn. However, I find it difficult to save weights in tensor flow's form and reuse it with other simulator softwares. 👀

Tks!

huigeopencv commented 1 year ago

Is the problem solved？I have the same problem