search

jbloomAus / DecisionTransformerInterpretability

Interpreting how transformers simulate agents performing RL tasks
https://jbloomaus-decisiontransformerinterpretability-app-4edcnc.streamlit.app/
MIT License
61 stars 15 forks source link

Fixed ablation tests #83

Closed JayBaileyCS closed 1 year ago

JayBaileyCS commented 1 year ago

Fixed the ablation tests by standardising names between heads_to_ablate and head_to_ablate. Decided on head_to_ablate since the tests pass an int.

Also, this is to test that I can get something through the whole lifecycle.