phess2 commited on
Commit
a03435c
Β·
1 Parent(s): 768c888

Fix dataset info for MLP models

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 ImageNet 1K
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 ImageNet 1K that were used to generate **Figure 3** in our paper.
33
 
34
  **Training Details:**
35
- - Dataset: ImageNet 1K
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