IsaacLab icon indicating copy to clipboard operation
IsaacLab copied to clipboard

[Question] Official pre-trained model checkpoints

Open j3soon opened this issue 1 year ago • 1 comments

Hi, I wonder if there's a place where we can access all pre-trained model weights (the .pt files) of the Isaac Lab environments?

So far, I have only found the pre-trained weights of ANYmal-C on Nucleus:

image

and some model checkpoints provided by SKRL on HuggingFace:

https://skrl.readthedocs.io/en/latest/intro/examples.html#isaac-lab-environments

Could you please let me know if there is a specific path where all the pre-trained checkpoints are stored, or if they are indeed not provided?

Thank you!

j3soon avatar Jul 29 '24 14:07 j3soon

Currently, we don't provide checkpoints for the trained models. The ANYmal one is uploaded since it is used for one of the other tasks in the framework. We can look into uploading them all in the future.

Will keep this unresolved as a reminder.

Mayankm96 avatar Jul 30 '24 08:07 Mayankm96

We now have pretrained checkpoints on Nucleus, such as http://omniverse-content-production.s3-us-west-2.amazonaws.com/Assets/Isaac/4.5/Isaac/IsaacLab/PretrainedCheckpoints/rsl_rl/Isaac-Cartpole-v0/checkpoint.pt

This can be downloaded by the --use_pretrained_checkpoint flag.

Ref: https://isaac-sim.github.io/IsaacLab/main/source/overview/reinforcement-learning/rl_existing_scripts.html

j3soon avatar Apr 27 '25 16:04 j3soon