Abhiram4 commited on
Commit
5b3802b
1 Parent(s): 361a3bd

Model save

Browse files
Files changed (2) hide show
  1. README.md +26 -19
  2. model.safetensors +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ base_model: microsoft/swinv2-tiny-patch4-window8-256
4
  tags:
5
  - generated_from_trainer
6
  datasets:
7
- - image_folder
8
  metrics:
9
  - accuracy
10
  model-index:
@@ -14,15 +14,15 @@ model-index:
14
  name: Image Classification
15
  type: image-classification
16
  dataset:
17
- name: image_folder
18
- type: image_folder
19
  config: default
20
- split: train
21
  args: default
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
- value: 0.9615316328342309
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
30
 
31
  # swinv2-tiny-patch4-window8-256-finetuned-eurosat
32
 
33
- This model is a fine-tuned version of [microsoft/swinv2-tiny-patch4-window8-256](https://huggingface.co/microsoft/swinv2-tiny-patch4-window8-256) on the image_folder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.1067
36
- - Accuracy: 0.9615
37
 
38
  ## Model description
39
 
@@ -53,28 +53,35 @@ More information needed
53
 
54
  The following hyperparameters were used during training:
55
  - learning_rate: 5e-05
56
- - train_batch_size: 32
57
- - eval_batch_size: 32
58
  - seed: 42
59
  - gradient_accumulation_steps: 4
60
- - total_train_batch_size: 128
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
63
  - lr_scheduler_warmup_ratio: 0.1
64
- - num_epochs: 3
65
 
66
  ### Training results
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
- | 0.2183 | 1.0 | 1188 | 0.1283 | 0.9542 |
71
- | 0.2099 | 2.0 | 2376 | 0.1188 | 0.9570 |
72
- | 0.1668 | 3.0 | 3564 | 0.1067 | 0.9615 |
 
 
 
 
 
 
 
73
 
74
 
75
  ### Framework versions
76
 
77
- - Transformers 4.33.0
78
- - Pytorch 2.0.0
79
- - Datasets 2.1.0
80
- - Tokenizers 0.13.3
 
4
  tags:
5
  - generated_from_trainer
6
  datasets:
7
+ - imagefolder
8
  metrics:
9
  - accuracy
10
  model-index:
 
14
  name: Image Classification
15
  type: image-classification
16
  dataset:
17
+ name: imagefolder
18
+ type: imagefolder
19
  config: default
20
+ split: validation
21
  args: default
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.983
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  # swinv2-tiny-patch4-window8-256-finetuned-eurosat
32
 
33
+ This model is a fine-tuned version of [microsoft/swinv2-tiny-patch4-window8-256](https://huggingface.co/microsoft/swinv2-tiny-patch4-window8-256) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.0482
36
+ - Accuracy: 0.983
37
 
38
  ## Model description
39
 
 
53
 
54
  The following hyperparameters were used during training:
55
  - learning_rate: 5e-05
56
+ - train_batch_size: 450
57
+ - eval_batch_size: 450
58
  - seed: 42
59
  - gradient_accumulation_steps: 4
60
+ - total_train_batch_size: 1800
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
63
  - lr_scheduler_warmup_ratio: 0.1
64
+ - num_epochs: 10
65
 
66
  ### Training results
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
+ | 0.3991 | 1.0 | 46 | 0.2074 | 0.933 |
71
+ | 0.1629 | 2.0 | 92 | 0.0946 | 0.971 |
72
+ | 0.1294 | 3.0 | 138 | 0.0692 | 0.977 |
73
+ | 0.1164 | 4.0 | 184 | 0.0572 | 0.982 |
74
+ | 0.1028 | 5.0 | 230 | 0.0494 | 0.984 |
75
+ | 0.0893 | 6.0 | 276 | 0.0487 | 0.982 |
76
+ | 0.0843 | 7.0 | 322 | 0.0472 | 0.984 |
77
+ | 0.0805 | 8.0 | 368 | 0.0437 | 0.983 |
78
+ | 0.0705 | 9.0 | 414 | 0.0523 | 0.982 |
79
+ | 0.0712 | 10.0 | 460 | 0.0482 | 0.983 |
80
 
81
 
82
  ### Framework versions
83
 
84
+ - Transformers 4.35.0
85
+ - Pytorch 2.1.0+cu121
86
+ - Datasets 2.14.6
87
+ - Tokenizers 0.14.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:726a1ce1d5f472fc494c05668bda4894b402ac4514a25ca6bfab437fb718957a
3
  size 110356296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e02b84225255abc6c662527c89669ed7cbd1fdb723437f34ead9c17328fa9f17
3
  size 110356296