Upload ModelConfiguration.py with huggingface_hub

ModelConfiguration.py

@@ -0,0 +1,13 @@
+
+from transformers import PretrainedConfig
+class VesselSegmentConfig(PretrainedConfig):
+  model_type = "SegformerForSemanticSegmentation"
+  def __init__(self, num_classes=1, input_channels=1, image_size=[512,512], features=[64,64,128], attention_dims=[64,32,16], patch_size=256, batch_size=4, **kwargs):
+    self.num_classes = num_classes
+    self.input_channels = input_channels
+    self.image_size = image_size
+    self.features = features
+    self.attention_dims = attention_dims
+    self.patch_size = patch_size
+    self.batch_size = batch_size
+    super().__init__(**kwargs)