/Pretrained-MoE-IDT

In-Context Soft-MoE Decision-Pretrained Transformer

Primary LanguagePython

This is code for paper "In-Context Decision Transformer: Reinforcement Learning via Hierarchical Chain-of-Thought"