Accelerator in MATLAB Reinforcement Learning Toolbox
조회 수: 3(최근 30일)
표시 이전 댓글
greencactus 2021년 7월 8일
답변: Drew Davis 2021년 7월 19일
Hello dear community,
I need to know if it's possible to work with accelerator like Jetson Nano and Intel Neural Compute Sticks in Reinforcement Learning Toolbox in the same way as in Deep Learning Toolbox.
I could not find the information in the documentation of the toolboxes. The RL Toolbox requires the DL Toolbox. But I have not find out if all layers and nets are saved in the DL Toolbox and further if the accelrator support packages like Simulink Code Support Package for NVIDIA Jetson CPUs supports the RL Toolboxes either.
Espectially I am interested in the RL examples:
1. Train PPO Agent for Automatic Parking Valet: https://de.mathworks.com/help/reinforcement-learning/ug/train-ppo-agent-for-automatic-parking-valet.html
2. Avoid Obstacles Using Reinforcement Learning for Mobile Robots: https://de.mathworks.com/help/robotics/ug/avoid-obstacles-using-reinforcement-learning-for-mobile-robots.html
Drew Davis 2021년 7월 19일
The acceleration features of the DL Toolbox support packages (e.g. Jetson CPU support for Simulink Coder) will accelerate predictions on the hardware. This is true if your network is trained with Deep Learning Toolbox training (trainNetwork) or Reinforcement Learning Toolbox training (train), however once finished with RL training, users will typically use generatePolicyFunction to create a policy function that is deployable to hardware.
Hope this helps
Find more on Deep Learning Toolbox in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!Start Hunting!