dictionary_training

Light-weight code training SAEs using the saprmarks/dictionary_learning library.

1. Execute setup.sh to setup the environment
2. Adapt training parameters in training_sweep.py
3. Run the training by executing run_training_sweep.py