Hugging Face's logo

hoanbklucky
/
dinov2-small-imagenet1k-1-layer-finetuned-noh

Image Classification
Transformers
TensorBoard
Safetensors
dinov2
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
hoanbklucky commited on
Commit
f0794af
•
1 Parent(s): 7067342

Training in progress, epoch 1

Browse files
Files changed (4) hide show
  1. config.json +57 -57
  2. model.safetensors +1 -1
  3. preprocessor_config.json +27 -27
  4. training_args.bin +1 -1
config.json CHANGED
@@ -1,57 +1,57 @@
1
- {
2
- "_name_or_path": "facebook/dinov2-small-imagenet1k-1-layer",
3
- "apply_layernorm": true,
4
- "architectures": [
5
- "Dinov2ForImageClassification"
6
- ],
7
- "attention_probs_dropout_prob": 0.0,
8
- "drop_path_rate": 0.0,
9
- "hidden_act": "gelu",
10
- "hidden_dropout_prob": 0.0,
11
- "hidden_size": 384,
12
- "id2label": {
13
- "0": "normal",
14
- "1": "cancer"
15
- },
16
- "image_size": 518,
17
- "initializer_range": 0.02,
18
- "label2id": {
19
- "cancer": 1,
20
- "normal": 0
21
- },
22
- "layer_norm_eps": 1e-06,
23
- "layerscale_value": 1.0,
24
- "mlp_ratio": 4,
25
- "model_type": "dinov2",
26
- "num_attention_heads": 6,
27
- "num_channels": 3,
28
- "num_hidden_layers": 12,
29
- "out_features": [
30
- "stage12"
31
- ],
32
- "out_indices": [
33
- 12
34
- ],
35
- "patch_size": 14,
36
- "problem_type": "single_label_classification",
37
- "qkv_bias": true,
38
- "reshape_hidden_states": true,
39
- "stage_names": [
40
- "stem",
41
- "stage1",
42
- "stage2",
43
- "stage3",
44
- "stage4",
45
- "stage5",
46
- "stage6",
47
- "stage7",
48
- "stage8",
49
- "stage9",
50
- "stage10",
51
- "stage11",
52
- "stage12"
53
- ],
54
- "torch_dtype": "float32",
55
- "transformers_version": "4.46.3",
56
- "use_swiglu_ffn": false
57
- }
 
1
+ {
2
+ "_name_or_path": "facebook/dinov2-small-imagenet1k-1-layer",
3
+ "apply_layernorm": true,
4
+ "architectures": [
5
+ "Dinov2ForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "drop_path_rate": 0.0,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 384,
12
+ "id2label": {
13
+ "0": "normal",
14
+ "1": "cancer"
15
+ },
16
+ "image_size": 518,
17
+ "initializer_range": 0.02,
18
+ "label2id": {
19
+ "cancer": 1,
20
+ "normal": 0
21
+ },
22
+ "layer_norm_eps": 1e-06,
23
+ "layerscale_value": 1.0,
24
+ "mlp_ratio": 4,
25
+ "model_type": "dinov2",
26
+ "num_attention_heads": 6,
27
+ "num_channels": 3,
28
+ "num_hidden_layers": 12,
29
+ "out_features": [
30
+ "stage12"
31
+ ],
32
+ "out_indices": [
33
+ 12
34
+ ],
35
+ "patch_size": 14,
36
+ "problem_type": "single_label_classification",
37
+ "qkv_bias": true,
38
+ "reshape_hidden_states": true,
39
+ "stage_names": [
40
+ "stem",
41
+ "stage1",
42
+ "stage2",
43
+ "stage3",
44
+ "stage4",
45
+ "stage5",
46
+ "stage6",
47
+ "stage7",
48
+ "stage8",
49
+ "stage9",
50
+ "stage10",
51
+ "stage11",
52
+ "stage12"
53
+ ],
54
+ "torch_dtype": "float32",
55
+ "transformers_version": "4.47.0",
56
+ "use_swiglu_ffn": false
57
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b3cd3ab042248ae03f2cff007982e01a36efb06dc987b8cb67100c83f6a57a8d
3
  size 88257824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd69cdb614c5934e64eda6481b398cc72dd7c9cc0a3c453741d53c0f3cdf7d3d
3
  size 88257824
preprocessor_config.json CHANGED
@@ -1,27 +1,27 @@
1
- {
2
- "crop_size": {
3
- "height": 224,
4
- "width": 224
5
- },
6
- "do_center_crop": true,
7
- "do_convert_rgb": true,
8
- "do_normalize": true,
9
- "do_rescale": true,
10
- "do_resize": true,
11
- "image_mean": [
12
- 0.485,
13
- 0.456,
14
- 0.406
15
- ],
16
- "image_processor_type": "BitImageProcessor",
17
- "image_std": [
18
- 0.229,
19
- 0.224,
20
- 0.225
21
- ],
22
- "resample": 3,
23
- "rescale_factor": 0.00392156862745098,
24
- "size": {
25
- "shortest_edge": 256
26
- }
27
- }
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_convert_rgb": true,
8
+ "do_normalize": true,
9
+ "do_rescale": true,
10
+ "do_resize": true,
11
+ "image_mean": [
12
+ 0.485,
13
+ 0.456,
14
+ 0.406
15
+ ],
16
+ "image_processor_type": "BitImageProcessor",
17
+ "image_std": [
18
+ 0.229,
19
+ 0.224,
20
+ 0.225
21
+ ],
22
+ "resample": 3,
23
+ "rescale_factor": 0.00392156862745098,
24
+ "size": {
25
+ "shortest_edge": 256
26
+ }
27
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3dfee905429128f44466b1f58efdace785116a9f8872c202e7495322ac7b1f90
3
  size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c10b3035031810b431b529a37c0fa4fd15f7b1d0861e51ca9195facbca9a3a4
3
  size 5368