mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-28 04:06:00 +00:00
docs(env_processor): add EnvConfig subclass step and policy_cfg examples
- Add missing '### 2. Update Your EnvConfig Subclass' section with get_env_processors() snippet - Update factory usage example to show policy_cfg parameter and keyword-argument style for both SmolVLA and ACT cases Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -90,11 +90,17 @@ The same policy can work with different environment processors, and the same env
|
|||||||
|
|
||||||
```python
|
```python
|
||||||
# Use SmolVLA policy with LIBERO environment
|
# Use SmolVLA policy with LIBERO environment
|
||||||
libero_preprocessor, libero_postprocessor = make_env_pre_post_processors(libero_cfg)
|
# Use SmolVLA policy with LIBERO environment
|
||||||
|
libero_preprocessor, libero_postprocessor = make_env_pre_post_processors(
|
||||||
|
env_cfg=libero_cfg,
|
||||||
|
policy_cfg=smolvla_cfg,
|
||||||
|
)
|
||||||
smolvla_preprocessor, smolvla_postprocessor = make_pre_post_processors(smolvla_cfg)
|
smolvla_preprocessor, smolvla_postprocessor = make_pre_post_processors(smolvla_cfg)
|
||||||
|
|
||||||
# Or use ACT policy with the same LIBERO environment
|
# Or use ACT policy with the same LIBERO environment
|
||||||
libero_preprocessor, libero_postprocessor = make_env_pre_post_processors(libero_cfg)
|
libero_preprocessor, libero_postprocessor = make_env_pre_post_processors(
|
||||||
|
env_cfg=libero_cfg,
|
||||||
|
policy_cfg=act_cfg,
|
||||||
|
)
|
||||||
act_preprocessor, act_postprocessor = make_pre_post_processors(act_cfg)
|
act_preprocessor, act_postprocessor = make_pre_post_processors(act_cfg)
|
||||||
```
|
```
|
||||||
|
|
||||||
@@ -328,6 +334,22 @@ class MyEnvConfig(EnvConfig):
|
|||||||
)
|
)
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### 2. Update Your `EnvConfig` Subclass
|
||||||
|
|
||||||
|
```python
|
||||||
|
# In src/lerobot/envs/configs.py
|
||||||
|
@EnvConfig.register_subclass("myenv")
|
||||||
|
@dataclass
|
||||||
|
class MyEnvConfig(EnvConfig):
|
||||||
|
# ... task/features/gym kwargs ...
|
||||||
|
def get_env_processors(self):
|
||||||
|
from lerobot.processor.pipeline import PolicyProcessorPipeline
|
||||||
|
return (
|
||||||
|
PolicyProcessorPipeline(steps=[MyEnvProcessorStep()]),
|
||||||
|
PolicyProcessorPipeline(steps=[]),
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
### 3. Use in Evaluation
|
### 3. Use in Evaluation
|
||||||
|
|
||||||
No changes needed! The evaluation script automatically uses the appropriate processor:
|
No changes needed! The evaluation script automatically uses the appropriate processor:
|
||||||
|
|||||||
Reference in New Issue
Block a user