Thabet commited on
Commit
58d9562
1 Parent(s): 4b6bd63

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -11
README.md CHANGED
@@ -17,10 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.1724
21
- - Accuracy: 0.9323
22
- - F1: 0.9304
23
- - Log Loss: 2.3373
24
 
25
  ## Model description
26
 
@@ -49,19 +49,20 @@ The following hyperparameters were used during training:
49
  - lr_scheduler_type: linear
50
  - lr_scheduler_warmup_ratio: 0.1
51
  - num_epochs: 3
 
52
 
53
  ### Training results
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Log Loss |
56
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:--------:|
57
- | 0.2618 | 0.98 | 32 | 0.2220 | 0.9066 | 0.9033 | 3.2276 |
58
- | 0.1771 | 1.98 | 64 | 0.1711 | 0.9291 | 0.9271 | 2.4486 |
59
- | 0.1575 | 2.98 | 96 | 0.1724 | 0.9323 | 0.9304 | 2.3373 |
60
 
61
 
62
  ### Framework versions
63
 
64
- - Transformers 4.26.0
65
- - Pytorch 1.13.1+cu116
66
- - Datasets 2.9.0
67
- - Tokenizers 0.13.2
 
17
 
18
  This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.2231
21
+ - Accuracy: 0.9119
22
+ - F1: 0.9086
23
+ - Log Loss: 3.0422
24
 
25
  ## Model description
26
 
 
49
  - lr_scheduler_type: linear
50
  - lr_scheduler_warmup_ratio: 0.1
51
  - num_epochs: 3
52
+ - mixed_precision_training: Native AMP
53
 
54
  ### Training results
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Log Loss |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:--------:|
58
+ | 0.2538 | 0.98 | 32 | 0.2327 | 0.9055 | 0.9018 | 3.2647 |
59
+ | 0.1735 | 1.98 | 64 | 0.2029 | 0.9151 | 0.9122 | 2.9309 |
60
+ | 0.1562 | 2.98 | 96 | 0.2231 | 0.9119 | 0.9086 | 3.0422 |
61
 
62
 
63
  ### Framework versions
64
 
65
+ - Transformers 4.27.0.dev0
66
+ - Pytorch 1.11.0
67
+ - Datasets 2.1.0
68
+ - Tokenizers 0.12.1