search

MushroomRL / mushroom-rl

Python library for Reinforcement Learning.
MIT License
803 stars 145 forks source link

Can I save pt files after training? #151

Open Harry-maximum opened 2 months ago

Harry-maximum commented 2 months ago

Hi! I am using the locomujoco repo to train my own robot, but now we prefer the .pt files to load to run in a real robot, so how can I save the policy as a .pt file instead of the .msh file?

boris-il-forte commented 1 month ago

the mushroom file is a zip containing the torch network. You can access the torch network by unzipping the file. There's no support in mushroom for .pt files: indeed, msh files is exactly a method to integrate mushroom agents with torch and other libraries.