tests.end_end package#
Submodules#
tests.end_end.test_decision_transformer module#
- tests.end_end.test_decision_transformer.download_training_data() None#
uses gdown to get data
- tests.end_end.test_decision_transformer.offline_config()#
- tests.end_end.test_decision_transformer.test_clone_transformer(download_training_data, n_ctx)#
- tests.end_end.test_decision_transformer.test_decision_transformer(download_training_data, n_ctx, transformer_model_config, offline_config)#
- tests.end_end.test_decision_transformer.test_decision_transformer_activation_fn(download_training_data, activation_fn, transformer_model_config, offline_config)#
- tests.end_end.test_decision_transformer.test_decision_transformer_bespoke()#
- tests.end_end.test_decision_transformer.test_decision_transformer_gated_mlp(download_training_data, gated_mlp, transformer_model_config, offline_config)#
- tests.end_end.test_decision_transformer.test_decision_transformer_optimizer(download_training_data, optimizer, transformer_model_config, offline_config)#
- tests.end_end.test_decision_transformer.test_decision_transformer_preln(download_training_data, n_ctx, transformer_model_config, offline_config)#
- tests.end_end.test_decision_transformer.test_decision_transformer_scheduler(download_training_data, scheduler, transformer_model_config, offline_config)#
- tests.end_end.test_decision_transformer.test_decision_transformer_state_embedding(download_training_data, state_embedding_type, transformer_model_config, offline_config)#
- tests.end_end.test_decision_transformer.transformer_model_config()#
tests.end_end.test_ppo module#
- tests.end_end.test_ppo.test_ppo_runner()#
- tests.end_end.test_ppo.test_ppo_runner_lstm_model()#
- tests.end_end.test_ppo.test_ppo_runner_traj_model()#
- tests.end_end.test_ppo.test_ppo_runner_traj_model_memory()#