| 18 | |
| 19 | |
| 20 | class Backbone(nn.Module, ABC, Generic[T]): |
| 21 | cfg: T |
| 22 | |
| 23 | def __init__( |
| 24 | self, |
| 25 | cfg: T, |
| 26 | num_frames: int | None, |
| 27 | image_shape: tuple[int, int] | None, |
| 28 | ) -> None: |
| 29 | super().__init__() |
| 30 | self.cfg = cfg |
| 31 | self.num_frames = num_frames |
| 32 | self.image_shape = image_shape |
| 33 | |
| 34 | @abstractmethod |
| 35 | def forward(self, batch: Batch, flows: Flows) -> BackboneOutput: |
| 36 | pass |
nothing calls this directly
no outgoing calls
no test coverage detected