Small DINOv3-based pixel decoder
(
in_chans=256, out_chans=3, upscale_factor=4, **kwargs
)
| 164 | |
| 165 | # Factory functions for different model sizes |
| 166 | def dinov3_pixel_decoder_small( |
| 167 | in_chans=256, out_chans=3, upscale_factor=4, **kwargs |
| 168 | ): |
| 169 | """Small DINOv3-based pixel decoder""" |
| 170 | model = DinoV3PixelDecoder( |
| 171 | in_chans=in_chans, |
| 172 | out_chans=out_chans, |
| 173 | upscale_factor=upscale_factor, |
| 174 | embed_dim=384, |
| 175 | depth=12, |
| 176 | num_heads=6, |
| 177 | ffn_ratio=4, |
| 178 | **kwargs, |
| 179 | ) |
| 180 | return model |
| 181 | |
| 182 | |
| 183 | def dinov3_pixel_decoder_base( |
nothing calls this directly
no test coverage detected