search

med-air / DEX

[ICRA'23] Demonstration-Guided Reinforcement Learning with Efficient Exploration for Task Automation of Surgical Robot
MIT License
32 stars 4 forks source link

How to use the trained model? #1

Closed S3668 closed 1 year ago

S3668 commented 1 year ago

I have followed your steps for training, but how should I get test data and use the trained model for testing? Looking forward to your recovery

TaoHuang13 commented 1 year ago

Thanks for your question @S3668.

Actually, we have integrated the evaluation module into the WandB. The evaluation is launched on the environment during training, thus no test data will be used. The evaluation results can be found in both WandB and the local log directory (exp_local/#task_name/#algo_name/#d100/#seed/eval.csv).

You may follow the office guidance of WandB to register your WandB account for result visualization.

Please let me know if there are any further questions.