atuo commited on
Commit
5364b16
1 Parent(s): eb8f313

Training in progress, epoch 0

Browse files
config.json CHANGED
@@ -1,13 +1,14 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
 
3
  "architectures": [
4
- "ViTForImageClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
- "encoder_stride": 16,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
- "hidden_size": 768,
11
  "id2label": {
12
  "0": "DR",
13
  "1": "G",
@@ -15,9 +16,8 @@
15
  "3": "WD",
16
  "4": "other"
17
  },
18
- "image_size": 224,
19
  "initializer_range": 0.02,
20
- "intermediate_size": 3072,
21
  "label2id": {
22
  "DR": 0,
23
  "G": 1,
@@ -25,14 +25,51 @@
25
  "WD": 3,
26
  "other": 4
27
  },
28
- "layer_norm_eps": 1e-12,
29
- "model_type": "vit",
30
- "num_attention_heads": 12,
 
 
31
  "num_channels": 3,
32
- "num_hidden_layers": 12,
33
- "patch_size": 16,
 
 
 
 
 
 
34
  "problem_type": "single_label_classification",
35
  "qkv_bias": true,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
36
  "torch_dtype": "float32",
37
- "transformers_version": "4.36.2"
 
38
  }
 
1
  {
2
+ "_name_or_path": "facebook/dinov2-large",
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": 1024,
12
  "id2label": {
13
  "0": "DR",
14
  "1": "G",
 
16
  "3": "WD",
17
  "4": "other"
18
  },
19
+ "image_size": 518,
20
  "initializer_range": 0.02,
 
21
  "label2id": {
22
  "DR": 0,
23
  "G": 1,
 
25
  "WD": 3,
26
  "other": 4
27
  },
28
+ "layer_norm_eps": 1e-06,
29
+ "layerscale_value": 1.0,
30
+ "mlp_ratio": 4,
31
+ "model_type": "dinov2",
32
+ "num_attention_heads": 16,
33
  "num_channels": 3,
34
+ "num_hidden_layers": 24,
35
+ "out_features": [
36
+ "stage24"
37
+ ],
38
+ "out_indices": [
39
+ 24
40
+ ],
41
+ "patch_size": 14,
42
  "problem_type": "single_label_classification",
43
  "qkv_bias": true,
44
+ "reshape_hidden_states": true,
45
+ "stage_names": [
46
+ "stem",
47
+ "stage1",
48
+ "stage2",
49
+ "stage3",
50
+ "stage4",
51
+ "stage5",
52
+ "stage6",
53
+ "stage7",
54
+ "stage8",
55
+ "stage9",
56
+ "stage10",
57
+ "stage11",
58
+ "stage12",
59
+ "stage13",
60
+ "stage14",
61
+ "stage15",
62
+ "stage16",
63
+ "stage17",
64
+ "stage18",
65
+ "stage19",
66
+ "stage20",
67
+ "stage21",
68
+ "stage22",
69
+ "stage23",
70
+ "stage24"
71
+ ],
72
  "torch_dtype": "float32",
73
+ "transformers_version": "4.35.2",
74
+ "use_swiglu_ffn": false
75
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e96567d5a698fe395078487f536178c594712539ac50e4be2a9589e15603a2da
3
- size 343233204
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4f16d0a90b7c2f366de617fa7b4dfd5332ec29c5949956e63483af4a1d2b8fc6
3
+ size 1217567100
preprocessor_config.json CHANGED
@@ -1,22 +1,27 @@
1
  {
 
 
 
 
 
 
2
  "do_normalize": true,
3
  "do_rescale": true,
4
  "do_resize": true,
5
  "image_mean": [
6
- 0.5,
7
- 0.5,
8
- 0.5
9
  ],
10
- "image_processor_type": "ViTImageProcessor",
11
  "image_std": [
12
- 0.5,
13
- 0.5,
14
- 0.5
15
  ],
16
- "resample": 2,
17
  "rescale_factor": 0.00392156862745098,
18
  "size": {
19
- "height": 224,
20
- "width": 224
21
  }
22
  }
 
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
  }
runs/Jan22_22-37-20_424f285bbf23/events.out.tfevents.1705963057.424f285bbf23.6619.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be8cb0bf3dcd7a9063f9bd95033543d44fe4045d393a67a01586bbcf15785187
3
+ size 4980
runs/Jan22_22-37-54_424f285bbf23/events.out.tfevents.1705963085.424f285bbf23.6619.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a55cf65ca5c056dcbd8a5fee55ec92471c7c426cc42428c5a381fbc379b4e70f
3
+ size 12985
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e86f796c1e3b5f959f41580a83e4304522c568c21e78d5086d958eaedb79a86f
3
- size 4792
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b447bd1c48a551e59eb0af020b63b4b16fd1924c0d7ba6bb7de8cd4304e92bc
3
+ size 4664