Project structure of: lucidrains/q-transformer
__init__.pyTransformer-based RL architecture import.agent.pyTrains Q-Transformer agent in RL tasks, handles memories and epsilons.attend.pyAttention-based neural network class with dropout, flash attentionmocks.pyMock environments and datasets for Q-transformer models in reinforcement learning.optimizer.pySeparates parameters, gets Adam or AdamW optimizer.q_learner.pyRobotic Transformer Q-Learner for Reinforcement Learningq_robotic_transformer.pyQ-Transformer for robotic action prediction
README.mdInstall Q-Transformer model for scalable offline RL with diverse action decoding.setup.pySetup file for Python package "q-transformer" with details and deps.