Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Loading weights from huggingface gives size mismatch. #223

Open
utkarshp1161 opened this issue Nov 13, 2024 · 0 comments
Open

Loading weights from huggingface gives size mismatch. #223

utkarshp1161 opened this issue Nov 13, 2024 · 0 comments
Labels
bug Something isn't working

Comments

@utkarshp1161
Copy link

Description

I want to run the checkpoint available at https://huggingface.co/skrl/IsaacOrbit-Isaac-Lift-Franka-v0-PPO/tree/main

When I execute the script torch_lift_franka_ppo.py as mentioned here: https://skrl.readthedocs.io/en/latest/intro/examples.html

ERROR I get:
RuntimeError: Error(s) in loading state_dict for Shared:
size mismatch for net.0.weight: copying a param with shape torch.Size([256, 30]) from checkpoint, the shape in current model is torch.Size([256, 36]).

What skrl version are you using?

1.3.0

What ML framework/library version are you using?

PyTorch

Additional system information

Linux

@utkarshp1161 utkarshp1161 added the bug Something isn't working label Nov 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant