Hugging Face's logo

TrieuNguyen
/
chest_xray_pneumonia

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
TrieuNguyen commited on
Commit
8c2d9b7
•
1 Parent(s): b254578

End of training

Browse files
Files changed (3) hide show
  1. README.md +13 -6
  2. model.safetensors +1 -1
  3. runs/Dec30_01-48-35_88b8ee47d05b/events.out.tfevents.1703900915.88b8ee47d05b.734.1 +2 -2
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.3081
21
- - Accuracy: 0.8942
22
 
23
  ## Model description
24
 
@@ -46,15 +46,22 @@ The following hyperparameters were used during training:
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
- - num_epochs: 3
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 0.1991 | 0.99 | 81 | 0.4470 | 0.8510 |
56
- | 0.1574 | 2.0 | 163 | 0.3711 | 0.8926 |
57
- | 0.1068 | 2.98 | 243 | 0.3081 | 0.8942 |
 
 
 
 
 
 
 
58
 
59
 
60
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.2508
21
+ - Accuracy: 0.9151
22
 
23
  ## Model description
24
 
 
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
+ - num_epochs: 10
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 0.1091 | 0.99 | 81 | 0.2422 | 0.9119 |
56
+ | 0.1085 | 2.0 | 163 | 0.2777 | 0.9167 |
57
+ | 0.1131 | 2.99 | 244 | 0.1875 | 0.9407 |
58
+ | 0.1129 | 4.0 | 326 | 0.2339 | 0.9183 |
59
+ | 0.0698 | 4.99 | 407 | 0.2581 | 0.9263 |
60
+ | 0.0904 | 6.0 | 489 | 0.2544 | 0.9167 |
61
+ | 0.0851 | 6.99 | 570 | 0.2023 | 0.9407 |
62
+ | 0.0833 | 8.0 | 652 | 0.2047 | 0.9327 |
63
+ | 0.0604 | 8.99 | 733 | 0.2738 | 0.9199 |
64
+ | 0.0671 | 9.94 | 810 | 0.2508 | 0.9151 |
65
 
66
 
67
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9e73e914d947071c40c79f15f7d322036ea4a6b65f61c8d08c43abac6d975f40
3
  size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9d9316a84fe88caae34191073967bdadea918dba28c1a3972ad0be0946ebc30
3
  size 343223968
runs/Dec30_01-48-35_88b8ee47d05b/events.out.tfevents.1703900915.88b8ee47d05b.734.1 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f3935abea40bf6ce29f14007359c7e60f4d129fd2241aa3bd2360efba32e5e3e
3
- size 20226
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2aa7f1b8f6fd6d039babc46098a863a47b17303022cc75ce97016475d9e40b8a
3
+ size 20580