Project structure of: lucidrains/q-transformer
__init__.py
Transformer-based RL architecture import.agent.py
Trains Q-Transformer agent in RL tasks, handles memories and epsilons.attend.py
Attention-based neural network class with dropout, flash attentionmocks.py
Mock environments and datasets for Q-transformer models in reinforcement learning.optimizer.py
Separates parameters, gets Adam or AdamW optimizer.q_learner.py
Robotic Transformer Q-Learner for Reinforcement Learningq_robotic_transformer.py
Q-Transformer for robotic action prediction
README.md
Install Q-Transformer model for scalable offline RL with diverse action decoding.setup.py
Setup file for Python package "q-transformer" with details and deps.