From 1e8bb2355bfe7c8d7c831c4a47abc26484831a15 Mon Sep 17 00:00:00 2001 From: TomorrowIsAnOtherDay <2466956298@qq.com> Date: Wed, 17 Jun 2020 17:26:27 +0800 Subject: [PATCH] add docs --- docs/zh_CN/Overview.md | 4 ++-- docs/zh_CN/tutorial/param.md | 21 +++++++++++++++++++++ 2 files changed, 23 insertions(+), 2 deletions(-) create mode 100644 docs/zh_CN/tutorial/param.md diff --git a/docs/zh_CN/Overview.md b/docs/zh_CN/Overview.md index 88f093b..f7866e3 100644 --- a/docs/zh_CN/Overview.md +++ b/docs/zh_CN/Overview.md @@ -42,8 +42,8 @@ diff --git a/docs/zh_CN/tutorial/param.md b/docs/zh_CN/tutorial/param.md new file mode 100644 index 0000000..a4acab4 --- /dev/null +++ b/docs/zh_CN/tutorial/param.md @@ -0,0 +1,21 @@ +# **教程:模型参数管理** +场景1: 在训练过程中,我们有时候需要把训练好的模型参数保存到本地,用于后续的部署或者评估。 + +当用户构建好agent之后,可以直接通过agent的相关接口来完成参数的存储。 +```python +agent = AtariAgent() +# 保存参数到 ./model.ckpt +agent.save('./model.ckpt') +# 恢复参数到这个agent上 +agent.restore('./model.ckpt') + +``` + +场景2: 并行训练过程中,经常需要把最新的模型参数同步到另一台服务器上,这时候,需要把模型参数拿到。 + +```python +#--------------Agent--------------- +weights = agent.get_weights() +#--------------Remote Actor-------------- +actor.set_weights(weights) +``` \ No newline at end of file -- GitLab