Update README.md

README.md

@@ -27,15 +27,15 @@ See `model/*/cp_*/cfg.json` for the hyperparameters that were used to train a pa
 ## Best Models:
 
 The best models for each of the model type are stored in the following directory:
-- DRC(3, 3):  `drc33/bkynosqi/cp_2002944000`
-- DRC(1, 1):  `drc11/eue6pax7/cp_2002944000`
-- ResNet:  `resnet/syb50iz7/cp_2002944000`
+| Model | Directory | Parameter Count |
+|------------|-----------|-----------------|
+| DRC(3, 3)  | `drc33/bkynosqi/cp_2002944000` | 1,285,125 (1.29M) |
+| DRC(1, 1)  | `drc11/eue6pax7/cp_2002944000` | 987,525 (0.99M)  |
+| ResNet     | `resnet/syb50iz7/cp_2002944000` | 3,068,421 (3.07M) |
 
-## Parameter counts:
+## Probes & SAEs:
 
-- DRC(3, 3):  1,285,125 (1.29M)
-- DRC(1, 1):  987,525 (0.99M)
-- ResNet:  3,068,421 (3.07M)
+The trained probes and SAEs are stored in the `probes` and `saes` directories, respectively.
 
 ## Training dataset:
 
@@ -43,7 +43,7 @@ The [Boxoban set of levels by DeepMind](https://github.com/google-deepmind/boxob
 
 # Citation
 
-If you use these neural networks, please cite our work:
+If you use any of these artifacts, please cite our work:
 
 ```bibtex
 @inproceedings{garriga-alonso2024planning,