| from transformers import PretrainedConfig | |
| class TimesformerScrollprizeConfig(PretrainedConfig): | |
| # model_type = "timesformerscrollprize" | |
| def __init__( | |
| self, | |
| window_size=64, | |
| depth=8, | |
| n_heads=6, | |
| patch_size=16, | |
| num_frames=26, | |
| num_classes=16, | |
| dim=512, | |
| **kwargs, | |
| ): | |
| self.window_size=window_size | |
| self.depth=depth | |
| self.n_heads=n_heads | |
| self.patch_size=patch_size | |
| self.num_frames=num_frames | |
| self.num_classes=num_classes | |
| self.dim=dim | |
| super().__init__(**kwargs) |