| from transformers import PretrainedConfig | |
| class SegVolConfig(PretrainedConfig): | |
| model_type = "segvol" | |
| def __init__( | |
| self, | |
| test_mode=True, | |
| test_w_zoom=False, | |
| **kwargs, | |
| ): | |
| self.spatial_size = [32, 256, 256] | |
| self.patch_size = [4, 16, 16] | |
| self.test_mode = test_mode | |
| self.test_w_zoom = test_w_zoom | |
| super().__init__(**kwargs) | |