(self)
| 1291 | self.init_weights() |
| 1292 | |
| 1293 | def init_weights(self) -> None: |
| 1294 | self.x_embedder.init_weights() |
| 1295 | self.pos_embedder.reset_parameters() |
| 1296 | if self.extra_per_block_abs_pos_emb: |
| 1297 | self.extra_pos_embedder.reset_parameters() |
| 1298 | |
| 1299 | self.t_embedder[1].init_weights() |
| 1300 | for block in self.blocks: |
| 1301 | block.init_weights() |
| 1302 | |
| 1303 | self.final_layer.init_weights() |
| 1304 | self.t_embedding_norm.reset_parameters() |
| 1305 | |
| 1306 | def build_patch_embed(self) -> None: |
| 1307 | ( |
no test coverage detected