shi-labs/oneformer_ade20k_swin_tiny
Original50.7Mparameters
Fixture7.9Mparameters
Layer stacks5compared
Findings0to review
✓ Same architecture, smaller
Every module class, layer kind and checked property of the original is in the fixture.
Model tester (advisory)
The model's own small test config in its ModelTester (10 literal values, 10 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.
model.pixel_level_module.encoder.swin.encoder.layers
Original · 4
Fixture · 4
| Kind | Original | Fixture | |
|---|---|---|---|
| SwinPatchMerging | 3 | 3 | ✓ kept |
| no SwinPatchMerging | 1 | 1 | ✓ kept |
model.pixel_level_module.decoder.input_projections
Original · 3
Fixture · 3
| Kind | Original | Fixture | |
|---|---|---|---|
| Conv2d | 3 | 3 | ✓ kept |
model.pixel_level_module.decoder.encoder.layers
Original · 6
Fixture · 2
| Kind | Original | Fixture | |
|---|---|---|---|
| FormerPixelDecoderEncoderLayer | 6 | 2 | ✓ kept |
model.transformer_module.decoder.query_transformer.decoder.layers
Original · 2
Fixture · 2
| Kind | Original | Fixture | |
|---|---|---|---|
| FormerTransformerDecoderQueryTransformerDecoderLayer | 2 | 2 | ✓ kept |
model.transformer_module.decoder.layers
Original · 9
Fixture · 1
| Kind | Original | Fixture | |
|---|---|---|---|
| FormerTransformerDecoderLayer | 9 | 1 | ✓ kept |
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 |
|---|---|---|---|---|
| model.pixel_level_module.encoder.swin.encoder.layers | 27.5M | 54.2% | 4.9M | 61.7% |
| model.transformer_module.decoder.layers | 14.2M | 28.0% | 659.3K | 8.3% |
| model.pixel_level_module.decoder.encoder.layers | 4.4M | 8.7% | 630.8K | 8.0% |
| model.transformer_module.decoder.query_transformer.decoder.layers | 3.2M | 6.2% | 1.3M | 16.6% |
| model.pixel_level_module | 691.3K | 1.4% | 179.3K | 2.3% |
| model.pixel_level_module.decoder.input_projections | 346.4K | 0.7% | 115.8K | 1.5% |
| model.transformer_module | 342.2K | 0.7% | 105.6K | 1.3% |
| 2 others | 85.8K | 0.2% | 26.5K | 0.3% |
Preserved properties
| Config | Property | Original | Fixture | |
|---|---|---|---|---|
| model | model_type | oneformer | oneformer | ✓ kept |
| backbone_config | model_type | swin | swin | ✓ kept |
| backbone_config | tie_word_embeddings | True | True | ✓ kept |
| backbone_config | hidden_act | gelu | gelu | ✓ kept |
| backbone_config | image_size | 224 | 224 | ✓ kept |
| backbone_config | patch_size | 4 | 4 | ✓ kept |
| backbone_config | num_channels | 3 | 3 | ✓ kept |
Module classes
Instances of each class in the built model (prefix One omitted).
| Class | Original | Fixture |
|---|---|---|
| Linear | 175 | 55 |
| LayerNorm | 80 | 32 |
| Dropout | 48 | 16 |
| ReLU | 33 | 9 |
| MultiheadAttention | 13 | 5 |
| NonDynamicallyQuantizableLinear | 13 | 5 |
| GELUActivation | 12 | 4 |
| SwinAttention | 12 | 4 |
| SwinLayer | 12 | 4 |
| SwinMLP | 12 | 4 |
| SwinRelativePositionBias | 12 | 4 |
| SwinDropPath | 11 | 3 |
| ModuleList | 9 | 9 |
| FormerAttention | 9 | 1 |
| FormerTransformerDecoderCrossAttentionLayer | 9 | 1 |
| FormerTransformerDecoderFFNLayer | 9 | 1 |
| FormerTransformerDecoderLayer | 9 | 1 |
| FormerTransformerDecoderSelfAttentionLayer | 9 | 1 |
| Conv2d | 8 | 8 |
| Sequential | 7 | 7 |
| FormerPixelDecoderEncoderLayer | 6 | 2 |
| FormerPixelDecoderEncoderMultiscaleDeformableAttention | 6 | 2 |
| GroupNorm | 5 | 5 |
| PredictionBlock | 5 | 5 |
| SwinStage | 4 | 4 |
| Identity | 3 | 3 |
| SwinPatchMerging | 3 | 3 |
| Embedding | 2 | 2 |
| FormerMLPPredictionHead | 2 | 2 |
| FormerSinePositionEmbedding | 2 | 2 |
| FormerTransformerDecoderQueryTransformerDecoderLayer | 2 | 2 |
| ModuleDict | 1 | 1 |
| FormerForUniversalSegmentation | 1 | 1 |
| FormerHungarianMatcher | 1 | 1 |
| FormerLoss | 1 | 1 |
| FormerModel | 1 | 1 |
| FormerPixelDecoder | 1 | 1 |
| FormerPixelDecoderEncoderOnly | 1 | 1 |
| FormerPixelLevelModule | 1 | 1 |
| FormerTaskModel | 1 | 1 |
| FormerTransformerDecoder | 1 | 1 |
| FormerTransformerDecoderQueryTransformer | 1 | 1 |
| FormerTransformerDecoderQueryTransformerDecoder | 1 | 1 |
| FormerTransformerModule | 1 | 1 |
| SwinBackbone | 1 | 1 |
| SwinEmbeddings | 1 | 1 |
| SwinEncoder | 1 | 1 |
| SwinModel | 1 | 1 |
| SwinPatchEmbeddings | 1 | 1 |
Config changes
11 fields changed, 90 unchanged.
| Field | Original | Fixture |
|---|---|---|
| backbone_config.depths | [2, 2, 6, 2] | [1, 1, 1, 1] |
| backbone_config.embed_dim | 96 | 64 |
| backbone_config.hidden_size | 768 | 64 |
| backbone_config.num_heads | [3, 6, 12, 24] | [2, 4, 8, 16] |
| backbone_config.num_layers | 4 | 2 |
| conv_dim | 256 | 128 |
| decoder_layers | 10 | 2 |
| encoder_layers | 6 | 2 |
| hidden_dim | 256 | 128 |
| mask_dim | 256 | 128 |
| num_attention_heads | 8 | 4 |