File size: 705 Bytes
dc10849 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | ---
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)
```
|