(self, decoder_partial: partial)
| 72 | self.decoder = self.load_decoder(decoder) |
| 73 | |
| 74 | def load_decoder(self, decoder_partial: partial) -> nn.Module: |
| 75 | decoder = decoder_partial( |
| 76 | in_chans=self.in_chans, |
| 77 | img_size=self.encoder.img_size, |
| 78 | input_embed_dim=self.embed_dim, |
| 79 | patch_size=self.patch_size, |
| 80 | ) |
| 81 | return decoder |
| 82 | |
| 83 | def random_masking(self, x): |
| 84 | """ |