gezx1004 commited on
Commit
6faa2ac
·
verified ·
1 Parent(s): 7324ed9

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
+ "architectures": [
3
+ "OrganLensEmbedder"
4
+ ],
5
+ "model_type": "organlens",
6
+ "checkpoint_revision": 524999,
7
+ "embedding_dim": 1024,
8
+ "embedding_method": "mask_patch_area_weighted_all_slices",
9
+ "supported_organs": [
10
+ "spleen",
11
+ "kidneys",
12
+ "liver",
13
+ "stomach",
14
+ "pancreas",
15
+ "lung",
16
+ "esophagus",
17
+ "trachea",
18
+ "intestine",
19
+ "heart",
20
+ "aorta"
21
+ ],
22
+ "checkpoint_files": {
23
+ "teacher": "teacher_checkpoint.pth",
24
+ "ctrate_heads": "heads/organlens_ctrate_heads_524999.pt",
25
+ "radchest_heads": "heads/organlens_radchest_heads_524999.pt"
26
+ }
27
+ }