MHasanUnical commited on
Commit
3702cdc
·
verified ·
1 Parent(s): f35a15c

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +27 -0
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "MultiscaleInputUNet",
3
+ "num_classes": 1,
4
+ "input_channels": 1,
5
+ "image_size": [
6
+ 512,
7
+ 512
8
+ ],
9
+ "features": [
10
+ 64,
11
+ 64,
12
+ 128
13
+ ],
14
+ "attention_dims": [
15
+ 64,
16
+ 32,
17
+ 16
18
+ ],
19
+ "patch_size": 256,
20
+ "batch_size": 4,
21
+ "architectures": [
22
+ "BinarySegmentation"
23
+ ],
24
+ "auto_map": {
25
+ "AutoModel": "model.py:VesselSegmentModel"
26
+ }
27
+ }