rukundob451 commited on
Commit
a97aa4c
·
verified ·
1 Parent(s): e297b36

Model save

Browse files
Files changed (2) hide show
  1. README.md +92 -0
  2. model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: facebook/convnext-tiny-224
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - imagefolder
8
+ metrics:
9
+ - accuracy
10
+ model-index:
11
+ - name: convnext-tiny-224-finetuned-papsmear
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: imagefolder
18
+ type: imagefolder
19
+ config: default
20
+ split: train
21
+ args: default
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.7794117647058824
26
+ ---
27
+
28
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
29
+ should probably proofread and complete it, then remove this comment. -->
30
+
31
+ # convnext-tiny-224-finetuned-papsmear
32
+
33
+ This model is a fine-tuned version of [facebook/convnext-tiny-224](https://huggingface.co/facebook/convnext-tiny-224) on the imagefolder dataset.
34
+ It achieves the following results on the evaluation set:
35
+ - Loss: 0.6010
36
+ - Accuracy: 0.7794
37
+
38
+ ## Model description
39
+
40
+ More information needed
41
+
42
+ ## Intended uses & limitations
43
+
44
+ More information needed
45
+
46
+ ## Training and evaluation data
47
+
48
+ More information needed
49
+
50
+ ## Training procedure
51
+
52
+ ### Training hyperparameters
53
+
54
+ The following hyperparameters were used during training:
55
+ - learning_rate: 5e-05
56
+ - train_batch_size: 16
57
+ - eval_batch_size: 16
58
+ - seed: 42
59
+ - gradient_accumulation_steps: 4
60
+ - total_train_batch_size: 64
61
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
+ - lr_scheduler_type: linear
63
+ - lr_scheduler_warmup_ratio: 0.1
64
+ - num_epochs: 15
65
+
66
+ ### Training results
67
+
68
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
+ |:-------------:|:-------:|:----:|:---------------:|:--------:|
70
+ | 1.8151 | 0.9870 | 19 | 1.6491 | 0.3456 |
71
+ | 1.6104 | 1.9740 | 38 | 1.4322 | 0.4265 |
72
+ | 1.4002 | 2.9610 | 57 | 1.2286 | 0.5882 |
73
+ | 1.203 | 4.0 | 77 | 1.0559 | 0.6544 |
74
+ | 1.047 | 4.9870 | 96 | 0.9357 | 0.6765 |
75
+ | 0.9083 | 5.9740 | 115 | 0.8477 | 0.7279 |
76
+ | 0.8756 | 6.9610 | 134 | 0.7762 | 0.75 |
77
+ | 0.7853 | 8.0 | 154 | 0.7258 | 0.7647 |
78
+ | 0.7198 | 8.9870 | 173 | 0.7023 | 0.7574 |
79
+ | 0.7151 | 9.9740 | 192 | 0.6756 | 0.7574 |
80
+ | 0.7049 | 10.9610 | 211 | 0.6493 | 0.7574 |
81
+ | 0.6387 | 12.0 | 231 | 0.6256 | 0.7721 |
82
+ | 0.6387 | 12.9870 | 250 | 0.6295 | 0.7721 |
83
+ | 0.6233 | 13.9740 | 269 | 0.6033 | 0.7794 |
84
+ | 0.632 | 14.8052 | 285 | 0.6010 | 0.7794 |
85
+
86
+
87
+ ### Framework versions
88
+
89
+ - Transformers 4.44.0
90
+ - Pytorch 2.4.0
91
+ - Datasets 2.21.0
92
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5fcebe9c32e6330bd2421a34d534b1359a29ae68c8c677dccca3e55ee09023fe
3
  size 111320240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc789181c201b87a9e0e146eda9e6ae7eae8872567449d4999e3267c6f46637c
3
  size 111320240