huawei-noah / SMARTS

Scalable Multi-Agent RL Training School for Autonomous Driving
MIT License
954 stars 190 forks source link

[Help Request] Debugging Needs During Training and Visualizing Specific Episodes #2096

Closed l1xiao closed 1 year ago

l1xiao commented 1 year ago

High Level Description

Hello,

I have some debugging needs during the training process(e10_drive). For instance, I have implemented a reward item based on the distance to the boundary and I would like to verify its correctness. Additionally, I am interested in viewing the replays and corresponding logs of failed cases during training. These needs can be summarized as: How can I inspect specific episodes frame by frame, and perform visualization or logging on them?

Has this requirement been implemented? If so, could you provide an example? If not, how should I utilize the existing envision to select a specific episode(or rollout), print logs, and add visualization content?

I have another question. I noticed there is a "Record" button in Envision, but no files are saved after clicking it. How should I use this feature?

Thank you for your assistance.

Version

1.4.0

Operating System

No response

Problems

No response

Adaickalavan commented 1 year ago

Hi @l1xiao,

A simple method for precise frame-by-frame visualization and debugging would be to plot the agent's observation image at every iteration. The plotter can be included inside the reward function and inside the preprocess function.

For example code which includes plotter functions for debugging, see the following:

l1xiao commented 1 year ago

Thanks for your reply