jbloomAus / SAELens

Training Sparse Autoencoders on Language Models
https://jbloomaus.github.io/SAELens/
MIT License
453 stars 120 forks source link

Add tests to training/evals.py #82

Closed chanind closed 2 weeks ago

chanind commented 7 months ago

This is currently the biggest file in the training dir missing tests (aside from the toy model stuff), and is important for training. Current coverage: https://app.codecov.io/gh/jbloomAus/SAELens/blob/main/sae_lens%2Ftraining%2Fevals.py

jbloomAus commented 1 month ago

There's some tests now but we could do with more.