Hugging Face's logo

ClementP
/
FundusDRGrading-vit_base_patch16_384

Transformers
PyTorch
Safetensors
English
fundus
diabetic retinopathy
classification
Inference Endpoints
Model card Files Community
ClementP commited on
Commit
c14c994
·
verified ·
1 Parent(s): d3afc9d

Add model

Browse files
Files changed (2) hide show
  1. config.json +28 -19
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,24 +1,33 @@
1
  {
2
  "architecture": "vit_base_patch16_384",
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": "vit_base_patch16_384",
3
+ "num_classes": 1,
4
+ "num_features": 768,
5
+ "global_pool": "token",
6
+ "pretrained_cfg": {
7
+ "tag": "augreg_in21k_ft_in1k",
8
+ "custom_load": true,
9
+ "input_size": [
10
+ 3,
11
+ 384,
12
+ 384
 
 
 
 
13
  ],
14
+ "fixed_input_size": true,
15
+ "interpolation": "bicubic",
16
+ "crop_pct": 1.0,
17
+ "crop_mode": "center",
18
+ "mean": [
19
+ 0.5,
20
+ 0.5,
21
+ 0.5
22
+ ],
23
+ "std": [
24
+ 0.5,
25
+ 0.5,
26
+ 0.5
27
+ ],
28
+ "num_classes": 1000,
29
+ "pool_size": null,
30
+ "first_conv": "patch_embed.proj",
31
+ "classifier": "head"
32
  }
33
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e00a8ee538b020f6ed687b93cfe2f8c078d564000d119e12d4a88e584183ea19
3
- size 345756316
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ef75c451db848a52b88f7e47e22dcae33ef46aa0901eeeaf2258a66d16827d7
3
+ size 345755404