Hugging Face's logo

bombshelll
/
swin-brain-modality-classification

Image Classification
Transformers
Safetensors
swin
Generated from Trainer
Model card Files
xet
 Community
bombshelll commited on
Commit
b4d99e1
·
verified ·
1 Parent(s): c936363

Model save

Browse files
Files changed (2) hide show
  1. README.md +24 -10
  2. model.safetensors +1 -1
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  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.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0377
22
- - Accuracy: 1.0
23
 
24
  ## Model description
25
 
@@ -51,14 +51,28 @@ The following hyperparameters were used during training:
51
 
52
  ### Training results
53
 
54
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
55
- |:-------------:|:------:|:----:|:---------------:|:--------:|
56
- | No log | 0.9333 | 7 | 1.0901 | 0.6415 |
57
- | 1.2435 | 2.0 | 15 | 0.4266 | 0.8585 |
58
- | 0.5378 | 2.9333 | 22 | 0.1524 | 0.9528 |
59
- | 0.1747 | 4.0 | 30 | 0.0747 | 0.9717 |
60
- | 0.1747 | 4.9333 | 37 | 0.0865 | 0.9623 |
61
- | 0.0905 | 6.0 | 45 | 0.0377 | 1.0 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
62
 
63
 
64
  ### Framework versions
 
18
 
19
  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.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.1550
22
+ - Accuracy: 0.9110
23
 
24
  ## Model description
25
 
 
51
 
52
  ### Training results
53
 
54
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
55
+ |:-------------:|:-------:|:----:|:---------------:|:--------:|
56
+ | 1.2625 | 0.9756 | 10 | 1.0232 | 0.5890 |
57
+ | 0.8162 | 1.9512 | 20 | 0.5373 | 0.8562 |
58
+ | 0.3924 | 2.9268 | 30 | 0.2490 | 0.8904 |
59
+ | 0.2528 | 4.0 | 41 | 0.2716 | 0.8767 |
60
+ | 0.2311 | 4.9756 | 51 | 0.2250 | 0.8904 |
61
+ | 0.2013 | 5.9512 | 61 | 0.2154 | 0.9110 |
62
+ | 0.1955 | 6.9268 | 71 | 0.1769 | 0.9178 |
63
+ | 0.1743 | 8.0 | 82 | 0.1451 | 0.9247 |
64
+ | 0.1664 | 8.9756 | 92 | 0.1489 | 0.9178 |
65
+ | 0.1503 | 9.9512 | 102 | 0.1964 | 0.8973 |
66
+ | 0.1518 | 10.9268 | 112 | 0.1708 | 0.9110 |
67
+ | 0.1456 | 12.0 | 123 | 0.1499 | 0.9041 |
68
+ | 0.1307 | 12.9756 | 133 | 0.1584 | 0.9041 |
69
+ | 0.1546 | 13.9512 | 143 | 0.1673 | 0.9110 |
70
+ | 0.1282 | 14.9268 | 153 | 0.1546 | 0.9041 |
71
+ | 0.121 | 16.0 | 164 | 0.1631 | 0.9041 |
72
+ | 0.1311 | 16.9756 | 174 | 0.1486 | 0.9041 |
73
+ | 0.1259 | 17.9512 | 184 | 0.1643 | 0.9178 |
74
+ | 0.1254 | 18.9268 | 194 | 0.1569 | 0.9110 |
75
+ | 0.1039 | 19.5122 | 200 | 0.1550 | 0.9110 |
76
 
77
 
78
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:784a804658d97710764591442c3b0aa2378a325301fe16b3e4a30cffe53486a1
3
  size 110348984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87c76132a578f7693f5528715eaefa66d1e4f83ae232b24e64fc4cf7a2470e03
3
  size 110348984