timesformer_config.py

from transformers.configuration_utils import PretrainedConfig


class TimesformerConfig(PretrainedConfig):

    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)