from transformers import PretrainedConfig

class Resnet3DScrollprizeConfig(PretrainedConfig):
    model_type = "resnetscrollprize"

    def __init__(
        self,
        window_size=64,
        model_depth=50,
        n_classes=1039,
        num_layers=18,
        **kwargs,
    ):
        self.window_size=window_size
        self.model_depth=model_depth
        self.n_classes=n_classes
        self.num_layers=num_layers
        super().__init__(**kwargs)