CoolFace
Modelpublic

anna4142/hierarchical-decision-transformer-hopper

sourceHugging Facemitupdated 2y agoView on Hugging Face
0likes14downloads
Model Card

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")