| 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) |