File size: 872 Bytes
3eba9ef |
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 |
---
language: en
tags:
- decision-transformer
- reinforcement-learning
- hierarchical-learning
- hopper
license: mit
---
# Hierarchical Decision Transformer for Hopper
This model extends the Decision Transformer architecture with hierarchical clustering capabilities for improved long-horizon task performance on the Hopper-v3 environment.
## Model Description
- **Model Type:** Hierarchical Decision Transformer
- **Training Environment:** Hopper-v3
- **Input:** State observations (11 dimensions), desired returns
- **Output:** Actions (3 dimensions)
- **Architecture Features:**
- Hierarchical clustering head for subtask identification
- Subgoal-based weighting
- Multi-task capabilities
## Usage
```python
from transformers import DecisionTransformerModel
model = DecisionTransformerModel.from_pretrained("anna4142/hierarchical-decision-transformer")
```
|