JorgeGIT commited on
Commit
86af773
1 Parent(s): 7029aa1

Model save

Browse files
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  license: apache-2.0
3
- base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
  - generated_from_trainer
6
  datasets:
@@ -22,7 +22,7 @@ model-index:
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.9774436090225563
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
30
 
31
  # finetuned-Leukemia-cell
32
 
33
- This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.1073
36
- - Accuracy: 0.9774
37
 
38
  ## Model description
39
 
@@ -65,12 +65,12 @@ The following hyperparameters were used during training:
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
- | 0.0673 | 2.94 | 100 | 0.1144 | 0.9699 |
69
- | 0.0639 | 5.88 | 200 | 0.1431 | 0.9549 |
70
- | 0.0491 | 8.82 | 300 | 0.2439 | 0.9436 |
71
- | 0.0532 | 11.76 | 400 | 0.1389 | 0.9624 |
72
- | 0.0636 | 14.71 | 500 | 0.1537 | 0.9624 |
73
- | 0.0493 | 17.65 | 600 | 0.1073 | 0.9774 |
74
 
75
 
76
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
+ base_model: facebook/deit-base-distilled-patch16-224
4
  tags:
5
  - generated_from_trainer
6
  datasets:
 
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.9661654135338346
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  # finetuned-Leukemia-cell
32
 
33
+ This model is a fine-tuned version of [facebook/deit-base-distilled-patch16-224](https://huggingface.co/facebook/deit-base-distilled-patch16-224) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.0956
36
+ - Accuracy: 0.9662
37
 
38
  ## Model description
39
 
 
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 0.3464 | 2.94 | 100 | 0.2377 | 0.9286 |
69
+ | 0.153 | 5.88 | 200 | 0.2519 | 0.9060 |
70
+ | 0.1663 | 8.82 | 300 | 0.1753 | 0.9398 |
71
+ | 0.0962 | 11.76 | 400 | 0.2009 | 0.9398 |
72
+ | 0.0612 | 14.71 | 500 | 0.1348 | 0.9586 |
73
+ | 0.0341 | 17.65 | 600 | 0.0956 | 0.9662 |
74
 
75
 
76
  ### Framework versions
config.json CHANGED
@@ -1,7 +1,7 @@
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,
@@ -30,7 +30,7 @@
30
  "trico": "6"
31
  },
32
  "layer_norm_eps": 1e-12,
33
- "model_type": "vit",
34
  "num_attention_heads": 12,
35
  "num_channels": 3,
36
  "num_hidden_layers": 12,
 
1
  {
2
+ "_name_or_path": "facebook/deit-base-distilled-patch16-224",
3
  "architectures": [
4
+ "DeiTForImageClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
  "encoder_stride": 16,
 
30
  "trico": "6"
31
  },
32
  "layer_norm_eps": 1e-12,
33
+ "model_type": "deit",
34
  "num_attention_heads": 12,
35
  "num_channels": 3,
36
  "num_hidden_layers": 12,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:411fb8baf76f74b477a876b3be53846a6c2c51f9277cb00a81542425c237112f
3
- size 343239356
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6776e9b0b9e3c3eb96201cecbd325ce233c78bbcd9bb45487d90adf02d5aa37c
3
+ size 343245796
preprocessor_config.json CHANGED
@@ -1,22 +1,28 @@
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": "ViTFeatureExtractor",
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_normalize": true,
8
  "do_rescale": true,
9
  "do_resize": true,
10
+ "feature_extractor_type": "DeiTFeatureExtractor",
11
  "image_mean": [
12
+ 0.485,
13
+ 0.456,
14
+ 0.406
15
  ],
16
+ "image_processor_type": "DeiTFeatureExtractor",
17
  "image_std": [
18
+ 0.229,
19
+ 0.224,
20
+ 0.225
21
  ],
22
+ "resample": 3,
23
  "rescale_factor": 0.00392156862745098,
24
  "size": {
25
+ "height": 256,
26
+ "width": 256
27
  }
28
  }
runs/Dec01_12-42-32_610b2a9400b8/events.out.tfevents.1701434600.610b2a9400b8.3653.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:425883d09328f67c7ca2932d6f0ef0682cc3a4cb4b49db448c7eebde1f58f1f9
3
+ size 4470
runs/Dec01_12-47-24_610b2a9400b8/events.out.tfevents.1701434856.610b2a9400b8.3653.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84aeb018fa61188f54aa42a754bdf62e3e7b1dd8cb619e25408eca14e462a201
3
+ size 4470
runs/Dec01_12-51-12_610b2a9400b8/events.out.tfevents.1701435078.610b2a9400b8.3653.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de28241f969d8e9e2ab222b47f6e01c98e1f3c9d7c8e05741cc16a6e1907ba38
3
+ size 6010
runs/Dec01_13-00-42_610b2a9400b8/events.out.tfevents.1701435648.610b2a9400b8.3653.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11338c2f5dcc39ab731748527d58078e5596ffe6a94cc7e02ffc97dd57fa241d
3
+ size 17396
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f0c51a9692dffa79666ee4c2a799f0ef6e3fddf95c8240e6d000dc48049d3457
3
  size 4600
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dacbff944cbe4f5a09b4fac9f1c809f6343455d2e3919516f15e7fd6ecdc5c74
3
  size 4600