For the new simplified training, go to src:
- Generate data with
data_generator.py - Train the transformer lens model with
train.py(will save config and model in same data subdirectory) - I've put a few initial exploratory functions in
training.ipynb