Fix dataset info for MLP models
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ This repository contains the figure data and model checkpoints supporting our pa
|
|
| 18 |
β βββ requirements.txt # Python dependencies
|
| 19 |
β βββ README.md # Detailed usage instructions
|
| 20 |
βββ models/ # Pre-trained model checkpoints
|
| 21 |
-
βββ MLPs/ # MLP models trained on
|
| 22 |
βββ transformers/ # Transformer models trained on Shakespeare
|
| 23 |
```
|
| 24 |
|
|
@@ -29,10 +29,10 @@ The `figures/` directory contains CSV files with the processed data used to crea
|
|
| 29 |
## Model Checkpoints
|
| 30 |
|
| 31 |
### MLP Models (`models/MLPs/`)
|
| 32 |
-
Two MLP models trained on
|
| 33 |
|
| 34 |
**Training Details:**
|
| 35 |
-
- Dataset:
|
| 36 |
- Purpose: Figure 3 generation and analysis
|
| 37 |
- Architecture: Multi-layer perceptrons with spectral constraints
|
| 38 |
|
|
|
|
| 18 |
β βββ requirements.txt # Python dependencies
|
| 19 |
β βββ README.md # Detailed usage instructions
|
| 20 |
βββ models/ # Pre-trained model checkpoints
|
| 21 |
+
βββ MLPs/ # MLP models trained on CIFAR-10
|
| 22 |
βββ transformers/ # Transformer models trained on Shakespeare
|
| 23 |
```
|
| 24 |
|
|
|
|
| 29 |
## Model Checkpoints
|
| 30 |
|
| 31 |
### MLP Models (`models/MLPs/`)
|
| 32 |
+
Two MLP models trained on CIFAR-10 that were used to generate **Figure 3** in our paper.
|
| 33 |
|
| 34 |
**Training Details:**
|
| 35 |
+
- Dataset: CIFAR-10
|
| 36 |
- Purpose: Figure 3 generation and analysis
|
| 37 |
- Architecture: Multi-layer perceptrons with spectral constraints
|
| 38 |
|