File size: 311 Bytes
2fbf9d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from transformers import PretrainedConfig
class SegVolConfig(PretrainedConfig):
model_type = "segvol"
def __init__(
self,
**kwargs,
):
self.spatial_size = [32, 256, 256]
self.patch_size = [4, 16, 16]
self.test_mode = True
super().__init__(**kwargs)
|