Change learning rate of RL DDPG networks after 1st training

2 次查看（过去 30 天）
Abdul Basith Ashraf
Abdul Basith Ashraf 2020-10-14
评论： Jonathan Zea 2022-1-27
I trained my DDPG networks using a particular learning rate. Now I want to improve the network by using a lower learning rate.
How do I change the learning rate without losing my perviously trained results?
  1 个评论
Jonathan Zea
Jonathan Zea 2022-1-27
I think you can train with your initial learning rate for a while, save the agent using the saveAgent option, loading after, and then change the learning rate and restart training.
% telling matlab to save the agent from 10 episode in folder "myfolder".
opts = rlTrainingOptions(...
'SaveAgentDirectory', myfolder, 'SaveAgentValue', 10);
% training agent in environment
trainingInfo = train(agent, env, opts);
After training for a while...
agent = load(...) % load the trained agent
actor = agent.getActor;
actor.Options.LearnRate = 0.001; % your new learning rate
agent = agent.setActor(actor);
% retrain agent in environment
trainingInfo = train(agent, env, opts);

请先登录，再进行评论。

请先登录，再回答此问题。

回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

产品

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by