(self, max_id=25, repeat_length=20)
| 114 | |
| 115 | class PositionalID(torch.nn.Module): |
| 116 | def __init__(self, max_id=25, repeat_length=20): |
| 117 | super().__init__() |
| 118 | self.max_id = max_id |
| 119 | self.repeat_length = repeat_length |
| 120 | |
| 121 | def frame_id_to_position_id(self, frame_id): |
| 122 | if frame_id < self.max_id: |