Hugging Face's logo

Leeyuyu
/
swin-tiny-patch4-window7-224-finetunedo

Image Classification
Transformers
PyTorch
swin
Generated from Trainer
Inference Endpoints
Model card Files Community
Leeyuyu commited on
Commit
3da2fb7
•
1 Parent(s): 93eb329

Model save

Browse files
Files changed (2) hide show
  1. README.md +11 -11
  2. pytorch_model.bin +1 -1
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 [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on the imagefolder dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.5094
21
- - Roc Auc: 0.7430
22
 
23
  ## Model description
24
 
@@ -37,7 +37,7 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - learning_rate: 5e-06
41
  - train_batch_size: 32
42
  - eval_batch_size: 32
43
  - seed: 42
@@ -52,14 +52,14 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Roc Auc |
54
  |:-------------:|:-----:|:----:|:---------------:|:-------:|
55
- | No log | 1.0 | 3 | 0.5437 | 0.7647 |
56
- | No log | 2.0 | 6 | 0.5345 | 0.7647 |
57
- | No log | 3.0 | 9 | 0.5252 | 0.7647 |
58
- | 0.5245 | 4.0 | 12 | 0.5188 | 0.7647 |
59
- | 0.5245 | 5.0 | 15 | 0.5151 | 0.7647 |
60
- | 0.5245 | 6.0 | 18 | 0.5122 | 0.7430 |
61
- | 0.5211 | 7.0 | 21 | 0.5102 | 0.7430 |
62
- | 0.5211 | 8.0 | 24 | 0.5094 | 0.7430 |
63
 
64
 
65
  ### Framework versions
 
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 the imagefolder dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.3710
21
+ - Roc Auc: 0.8606
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 5e-05
41
  - train_batch_size: 32
42
  - eval_batch_size: 32
43
  - seed: 42
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Roc Auc |
54
  |:-------------:|:-----:|:----:|:---------------:|:-------:|
55
+ | No log | 1.0 | 3 | 0.5066 | 0.7647 |
56
+ | No log | 2.0 | 6 | 0.4204 | 0.7941 |
57
+ | No log | 3.0 | 9 | 0.4298 | 0.7353 |
58
+ | 0.4868 | 4.0 | 12 | 0.4040 | 0.8018 |
59
+ | 0.4868 | 5.0 | 15 | 0.3925 | 0.7724 |
60
+ | 0.4868 | 6.0 | 18 | 0.3674 | 0.8235 |
61
+ | 0.4096 | 7.0 | 21 | 0.3673 | 0.8606 |
62
+ | 0.4096 | 8.0 | 24 | 0.3710 | 0.8606 |
63
 
64
 
65
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:697986de8ecd3e26360ca4158f2109fb91dfc05bf4494e7326b8f0f86165c4b7
3
  size 110394865
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:189b4f6d91e51cc42eef5eb609599d928bcfd059790639a4ca25c5eb5fa05e8a
3
  size 110394865