(self, x: torch.Tensor)
| 799 | self.temperal_upsample = temperal_downsample[::-1] |
| 800 | |
| 801 | def _encode(self, x: torch.Tensor) -> torch.Tensor: |
| 802 | x = [ |
| 803 | self.model.encode(u.unsqueeze(0)).squeeze(0) |
| 804 | for u in x |
| 805 | ] |
| 806 | x = torch.stack(x) |
| 807 | return x |
| 808 | |
| 809 | @apply_forward_hook |
| 810 | def encode( |