(self, cfg: Config, *, parent: Module)
| 1477 | super().__init__(cfg, parent=parent) |
| 1478 | cfg = self.config |
| 1479 | self._add_child( |
| 1480 | "spatial", |
| 1481 | cfg.spatial_embeddings.set(dim=cfg.dim), |
| 1482 | ) |
| 1483 | self._add_child( |
| 1484 | "temporal", |
| 1485 | cfg.temporal_embeddings.set(dim=cfg.dim), |
| 1486 | ) |
| 1487 | |
| 1488 | @property |
| 1489 | def num_spatial_embeddings(self) -> int: |
| 1490 | return self.spatial.embeddings().shape[0] |
| 1491 | |
| 1492 | @property |
nothing calls this directly
no test coverage detected