Set gpu option for rlPPOAgent actor
4 visualizaciones (últimos 30 días)
Mostrar comentarios más antiguos
John Doe
el 10 de Mzo. de 2021
Comentada: Emmanouil Tzorakoleftherakis
el 25 de Mzo. de 2021
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!
0 comentarios
Respuestas (1)
Emmanouil Tzorakoleftherakis
el 11 de Mzo. de 2021
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 comentarios
Emmanouil Tzorakoleftherakis
el 25 de Mzo. de 2021
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
Ver también
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!