Nikitapyscript commited on
Commit
ef864d5
·
verified ·
1 Parent(s): dcb5824

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +48 -0
config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "monai_unet_3d",
3
+ "architecture": {
4
+ "spatial_dims": 3,
5
+ "in_channels": 1,
6
+ "out_channels": 2,
7
+ "channels": [
8
+ 16,
9
+ 32,
10
+ 64,
11
+ 128,
12
+ 256,
13
+ 512
14
+ ],
15
+ "strides": [
16
+ 2,
17
+ 2,
18
+ 2,
19
+ 2,
20
+ 2,
21
+ 2
22
+ ],
23
+ "num_res_units": 2,
24
+ "dropout": 0.2
25
+ },
26
+ "training": {
27
+ "patch_size": [
28
+ 256,
29
+ 256,
30
+ 256
31
+ ],
32
+ "loss_function": "DiceLoss",
33
+ "optimizer": "Adam",
34
+ "learning_rate": 0.0001,
35
+ "weight_decay": 1e-05
36
+ },
37
+ "preprocessing": {
38
+ "intensity_range": [
39
+ -1000,
40
+ 4000
41
+ ],
42
+ "output_range": [
43
+ 0,
44
+ 1
45
+ ],
46
+ "normalization": "ScaleIntensityRanged"
47
+ }
48
+ }