google/switch-base-8
Original619.3Mparameters
Fixture2.3Mparameters
Layer stacks2compared
Findings2to review
⚠ To review
- encoder.block: layer kind 'no Embedding' is not in the fixture
- decoder.block: layer kind 'no Embedding' is not in the fixture
Model tester (advisory)
The model's own small test config in its ModelTester (23 literal values, 23 set by the model's own tester, the rest generic base-tester defaults). Testers are usually representative but not always, so these notes are for review, not rules.
The fixture covers every per-layer kind the tester sets, and shrinks every size it does.
Layer stacks
One square per layer, colored by kind: the layer's module classes and type attributes. Hover a square for its index and kind.
encoder.block
Original · 12
Fixture · 2
| Kind | Original | Fixture | |
|---|---|---|---|
| Embedding | 1 | 1 | ✓ kept |
| Experts, SparseMLP, Top1Router | 6 | 1 | ✓ kept |
| no Embedding | 5 | 0 | ⚠ missing |
decoder.block
Original · 12
Fixture · 2
| Kind | Original | Fixture | |
|---|---|---|---|
| Embedding | 1 | 1 | ✓ kept |
| Experts, SparseMLP, Top1Router | 6 | 1 | ✓ kept |
| no Embedding | 5 | 0 | ⚠ missing |
Where the parameters are
Share of each model's parameters per component. Vocabulary and patch sizes are kept, so embeddings usually take a larger share of the fixture.
Original
Fixture
| Component | Original | Share | Fixture | Share |
|---|---|---|---|---|
| decoder.block | 311.5M | 50.3% | 148.2K | 6.4% |
| encoder.block | 283.2M | 45.7% | 115.3K | 5.0% |
| shared | 24.7M | 4.0% | 2.1M | 88.6% |
| decoder.final_layer_norm | 768 | 0.0% | 64 | 0.0% |
| encoder.final_layer_norm | 768 | 0.0% | 64 | 0.0% |
Preserved properties
| Config | Property | Original | Fixture | |
|---|---|---|---|---|
| model | model_type | switch_transformers | switch_transformers | ✓ kept |
| model | vocab_size | 32128 | 32128 | ✓ kept |
| model | tie_word_embeddings | True | True | ✓ kept |
Module classes
Instances of each class in the built model (prefix SwitchTransformers omitted).
| Class | Original | Fixture |
|---|---|---|
| Linear | 373 | 47 |
| Dropout | 170 | 22 |
| ReLU | 108 | 10 |
| DenseActDense | 108 | 10 |
| LayerNorm | 62 | 12 |
| Attention | 36 | 6 |
| ModuleList | 26 | 6 |
| Block | 24 | 4 |
| LayerFF | 24 | 4 |
| LayerSelfAttention | 24 | 4 |
| Experts | 12 | 2 |
| LayerCrossAttention | 12 | 2 |
| SparseMLP | 12 | 2 |
| Top1Router | 12 | 2 |
| Embedding | 5 | 5 |
| Stack | 2 | 2 |
| ForConditionalGeneration | 1 | 1 |
Config changes
7 fields changed, 42 unchanged.
| Field | Original | Fixture |
|---|---|---|
| d_ff | 3072 | 128 |
| d_kv | 64 | 16 |
| d_model | 768 | 64 |
| num_decoder_layers | 12 | 2 |
| num_experts | 8 | 4 |
| num_heads | 12 | 4 |
| num_layers | 12 | 2 |