Set gpu option for rlPPOAgent actor

23 次查看（过去 30 天）
John Doe
John Doe 2021-3-10
评论： Emmanouil Tzorakoleftherakis 2021-3-25
Hi,
Is there a way to set the gpu option for an rlPPOAgent actor?
agent = rlPPOAgent(obsInfo,actInfo,agentOpts);
critic = getCritic(agent);
critic.Options.LearnRate = 1e-3;
critic.Options.UseDevice = 'gpu';
agent = setCritic(agent,critic);
I was able to set the critic to gpu but need to know how to set it for the actor. agentOpts is rlPPOAgentOptions, which doesn't take UseDevice.
Thanks!

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

回答（1 个）

Emmanouil Tzorakoleftherakis
What you have specified is sufficient for the critic. If you do the same for the actor you are all set - there is no additional param for the agent overall (actor and critic are two key components of the agent after all)
  7 个评论
显示 5更早的评论
John Doe
John Doe 2021-3-25
I could use some guidance here to run fully on GPU. My env runs pretty slow right now
Emmanouil Tzorakoleftherakis
Emmanouil Tzorakoleftherakis 2021-3-25
I believe I already answered this question - for the actor you need to do the same thing you did for the critic.
actor = getActor(agent);
actor.Options.UseDevice = 'gpu';
agent = setCritic(agent,actor);
If the environment itse;f is simulating slowly GPUs will not help as much. You need to check your model and identify what's slow

请先登录，再进行评论。

类别

在 Help Center 和 File Exchange 中查找有关 Policies and Value Functions 的更多信息

标签

产品


版本

R2020b

Community Treasure Hunt

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

Start Hunting!

Translated by