Train your first Decision Transformer
📰 Hugging Face Blog
Train a Decision Transformer model from scratch to make a half-cheetah run using the transformers library
Action Steps
- Load the dataset and build a Custom Data Collator
- Train the Decision Transformer model using a transformers Trainer
- Fine-tune the model for a specific task, such as making a half-cheetah run
Who Needs to Know This
Machine learning engineers and AI researchers can benefit from this tutorial to learn how to train Decision Transformer models for offline decision-making tasks
Key Insight
💡 Decision Transformers can be used as decision-making models, and can be trained from scratch using the transformers library
Share This
🤖 Train your first Decision Transformer model to make a half-cheetah run! 💻
DeepCamp AI