Upload config.json with huggingface_hub
Browse files- config.json +1 -2
config.json
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
{
|
| 2 |
-
"model_type": "VesselSegmentModel",
|
| 3 |
"num_classes": 1,
|
| 4 |
"input_channels": 1,
|
| 5 |
"image_size": [
|
|
@@ -22,7 +21,7 @@
|
|
| 22 |
"BinarySegmentation"
|
| 23 |
],
|
| 24 |
"auto_map": {
|
| 25 |
-
"AutoConfig": "model.py:
|
| 26 |
"AutoModel": "model.py:VesselSegmentModel"
|
| 27 |
}
|
| 28 |
}
|
|
|
|
| 1 |
{
|
|
|
|
| 2 |
"num_classes": 1,
|
| 3 |
"input_channels": 1,
|
| 4 |
"image_size": [
|
|
|
|
| 21 |
"BinarySegmentation"
|
| 22 |
],
|
| 23 |
"auto_map": {
|
| 24 |
+
"AutoConfig": "model.py:VesselSegmentConfig",
|
| 25 |
"AutoModel": "model.py:VesselSegmentModel"
|
| 26 |
}
|
| 27 |
}
|