Hugging Face's logo

bombshelll
/
swin-brain-plane-classification

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

Model save

Browse files
Files changed (2) hide show
  1. README.md +15 -17
  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.0669
22
- - Accuracy: 0.9825
23
 
24
  ## Model description
25
 
@@ -39,11 +39,11 @@ More information needed
39
 
40
  The following hyperparameters were used during training:
41
  - learning_rate: 5e-05
42
- - train_batch_size: 32
43
- - eval_batch_size: 32
44
  - seed: 42
45
  - gradient_accumulation_steps: 4
46
- - total_train_batch_size: 128
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
49
  - lr_scheduler_warmup_ratio: 0.1
@@ -51,18 +51,16 @@ The following hyperparameters were used during training:
51
 
52
  ### Training results
53
 
54
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
55
- |:-------------:|:------:|:----:|:---------------:|:--------:|
56
- | 1.1297 | 0.9412 | 4 | 0.8378 | 0.7807 |
57
- | 0.7737 | 1.8824 | 8 | 0.4059 | 0.9298 |
58
- | 0.4514 | 2.8235 | 12 | 0.2630 | 0.9474 |
59
- | 0.2684 | 4.0 | 17 | 0.1095 | 0.9825 |
60
- | 0.2184 | 4.9412 | 21 | 0.1291 | 0.9737 |
61
- | 0.2021 | 5.8824 | 25 | 0.0686 | 0.9825 |
62
- | 0.2078 | 6.8235 | 29 | 0.0541 | 0.9825 |
63
- | 0.1107 | 8.0 | 34 | 0.0637 | 0.9825 |
64
- | 0.1373 | 8.9412 | 38 | 0.0691 | 0.9825 |
65
- | 0.1381 | 9.4118 | 40 | 0.0669 | 0.9825 |
66
 
67
 
68
  ### 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.4059
22
+ - Accuracy: 0.9091
23
 
24
  ## Model description
25
 
 
39
 
40
  The following hyperparameters were used during training:
41
  - learning_rate: 5e-05
42
+ - train_batch_size: 16
43
+ - eval_batch_size: 16
44
  - seed: 42
45
  - gradient_accumulation_steps: 4
46
+ - total_train_batch_size: 64
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
49
  - lr_scheduler_warmup_ratio: 0.1
 
51
 
52
  ### Training results
53
 
54
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
55
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
56
+ | 1.5674 | 0.8 | 2 | 1.1258 | 0.3636 |
57
+ | 0.8875 | 2.0 | 5 | 0.9127 | 0.7273 |
58
+ | 1.1485 | 2.8 | 7 | 0.7617 | 0.7879 |
59
+ | 0.6152 | 4.0 | 10 | 0.6071 | 0.8485 |
60
+ | 0.7519 | 4.8 | 12 | 0.5297 | 0.8788 |
61
+ | 0.4554 | 6.0 | 15 | 0.4559 | 0.9091 |
62
+ | 0.5892 | 6.8 | 17 | 0.4248 | 0.9091 |
63
+ | 0.4184 | 8.0 | 20 | 0.4059 | 0.9091 |
 
 
64
 
65
 
66
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:097aa4b874cf5919eab63bccf59613d95999d1e0a48fb9e2fc2790360ca37385
3
  size 110345908
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59c8887113e2a04c63f8efab31565b75634a66e897af3a7c92937e7ddb6188e0
3
  size 110345908