| library_name: openpi | |
| tags: | |
| - robotics | |
| - imitation-learning | |
| - policy-learning | |
| # OpenPI Fine-tuned Model | |
| This model was fine-tuned using OpenPI. | |
| ## Training Details | |
| - **Global Step**: 5000 | |
| - **Experiment Name**: droid-finetune-cube-stacking | |
| - **Learning Rate**: N/A | |
| - **Batch Size**: 32 | |
| ## Usage | |
| ```python | |
| from openpi.models_pytorch.pi0_pytorch import PI0Pytorch | |
| from safetensors.torch import load_file | |
| from huggingface_hub import hf_hub_download | |
| # Download model weights | |
| model_path = hf_hub_download(repo_id="your-repo-id", filename="model.safetensors") | |
| state_dict = load_file(model_path) | |
| # Load model (adjust config as needed) | |
| model = PI0Pytorch(...) | |
| model.load_state_dict(state_dict) | |
| ``` | |