Instructions to use qb1t/lekiwi-reward-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LeRobot
How to use qb1t/lekiwi-reward-classifier with LeRobot:
- Notebooks
- Google Colab
- Kaggle
| { | |
| "name": "classifier_preprocessor", | |
| "steps": [ | |
| { | |
| "registry_name": "normalizer_processor", | |
| "config": { | |
| "eps": 1e-08, | |
| "features": { | |
| "observation.images.front": { | |
| "type": "VISUAL", | |
| "shape": [ | |
| 3, | |
| 128, | |
| 128 | |
| ] | |
| }, | |
| "observation.images.wrist": { | |
| "type": "VISUAL", | |
| "shape": [ | |
| 3, | |
| 128, | |
| 128 | |
| ] | |
| } | |
| }, | |
| "norm_map": { | |
| "VISUAL": "MEAN_STD" | |
| } | |
| }, | |
| "state_file": "classifier_preprocessor_step_0_normalizer_processor.safetensors" | |
| }, | |
| { | |
| "registry_name": "normalizer_processor", | |
| "config": { | |
| "eps": 1e-08, | |
| "features": { | |
| "next.reward": { | |
| "type": "REWARD", | |
| "shape": [ | |
| 1 | |
| ] | |
| } | |
| }, | |
| "norm_map": { | |
| "VISUAL": "MEAN_STD" | |
| } | |
| }, | |
| "state_file": "classifier_preprocessor_step_1_normalizer_processor.safetensors" | |
| }, | |
| { | |
| "registry_name": "device_processor", | |
| "config": { | |
| "device": "mps", | |
| "float_dtype": null | |
| } | |
| } | |
| ] | |
| } |