Hugging Face's logo

ClementP
/
FundusDRGrading-tf_efficientnet_b5

Transformers
PyTorch
Safetensors
English
fundus
diabetic retinopathy
classification
Eval Results
Inference Endpoints
Model card Files Community
ClementP commited on
Commit
dbf95bf
•
1 Parent(s): f6927d2

Add model

Browse files
Files changed (2) hide show
  1. config.json +30 -19
  2. pytorch_model.bin +3 -0
config.json CHANGED
@@ -1,24 +1,35 @@
1
  {
2
  "architecture": "tf_efficientnet_b5",
3
- "as_regression": true,
4
- "ema": {
5
- "decay": 0.999
6
- },
7
- "img_size": [
8
- 512,
9
- 512
10
- ],
11
- "lr": 3e-05,
12
- "num_classes": 5,
13
- "optimizer": {
14
- "betas": [
15
- 0.9,
16
- 0.999
17
  ],
18
- "weight_decay": 0.0005
19
- },
20
- "pretrained": true,
21
- "swa": {
22
- "swa_lrs": 3e-05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
  }
24
  }
 
1
  {
2
  "architecture": "tf_efficientnet_b5",
3
+ "num_classes": 1,
4
+ "num_features": 2048,
5
+ "pretrained_cfg": {
6
+ "tag": "ns_jft_in1k",
7
+ "custom_load": false,
8
+ "input_size": [
9
+ 3,
10
+ 456,
11
+ 456
 
 
 
 
 
12
  ],
13
+ "fixed_input_size": false,
14
+ "interpolation": "bicubic",
15
+ "crop_pct": 0.934,
16
+ "crop_mode": "center",
17
+ "mean": [
18
+ 0.485,
19
+ 0.456,
20
+ 0.406
21
+ ],
22
+ "std": [
23
+ 0.229,
24
+ 0.224,
25
+ 0.225
26
+ ],
27
+ "num_classes": 1000,
28
+ "pool_size": [
29
+ 15,
30
+ 15
31
+ ],
32
+ "first_conv": "conv_stem",
33
+ "classifier": "classifier"
34
  }
35
  }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:849ddfe7e88fa2e733a1a953dddbdfa22a9f1bb4ad3d45e319c63e21e0233445
3
+ size 114353122