search

mihdalal / planseqlearn

[ICLR 2024] PyTorch Code for Plan-Seq-Learn: Language Model Guided RL for Solving Long Horizon Robotics Tasks
https://mihdalal.github.io/planseqlearn/
66 stars 4 forks source link

Running speed #5

Closed fighting-j closed 2 months ago

fighting-j commented 5 months ago

Hello, I installed PyTorch 1.12.1 to run planseqlearn/launch_scripts/psl_kitchen.py, but the gpu usage is very low and the training speed is very slow. Is there any way to speed it up?

mihdalal commented 5 months ago

What FPS are you getting? I think unfortunately the kitchen env is quite slow. You can try reducing the frameskip in the kitchen env to speed things up, but it may cause reduced precision in the end-effector control actions.

mihdalal commented 5 months ago

If you end up finding a way to speed things up, please make a pull request!

mihdalal commented 2 months ago

Closing due to inactivity.