matinbaig43 commited on
Commit
2159027
1 Parent(s): d0af7b5

Model save

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -23,7 +23,7 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.9777131782945736
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,8 +33,8 @@ should probably proofread and complete it, then remove this comment. -->
33
 
34
  This model is a fine-tuned version of [microsoft/swinv2-tiny-patch4-window16-256](https://huggingface.co/microsoft/swinv2-tiny-patch4-window16-256) on the imagefolder dataset.
35
  It achieves the following results on the evaluation set:
36
- - Loss: 0.0636
37
- - Accuracy: 0.9777
38
 
39
  ## Model description
40
 
@@ -59,7 +59,7 @@ The following hyperparameters were used during training:
59
  - seed: 42
60
  - gradient_accumulation_steps: 4
61
  - total_train_batch_size: 32
62
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
63
  - lr_scheduler_type: linear
64
  - lr_scheduler_warmup_ratio: 0.1
65
  - num_epochs: 1
@@ -68,12 +68,12 @@ The following hyperparameters were used during training:
68
 
69
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
70
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
71
- | 0.103 | 1.0 | 516 | 0.0636 | 0.9777 |
72
 
73
 
74
  ### Framework versions
75
 
76
- - Transformers 4.44.2
77
- - Pytorch 2.5.0+cu121
78
  - Datasets 3.1.0
79
- - Tokenizers 0.19.1
 
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
+ value: 0.9798934108527132
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
33
 
34
  This model is a fine-tuned version of [microsoft/swinv2-tiny-patch4-window16-256](https://huggingface.co/microsoft/swinv2-tiny-patch4-window16-256) on the imagefolder dataset.
35
  It achieves the following results on the evaluation set:
36
+ - Loss: 0.0568
37
+ - Accuracy: 0.9799
38
 
39
  ## Model description
40
 
 
59
  - seed: 42
60
  - gradient_accumulation_steps: 4
61
  - total_train_batch_size: 32
62
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
63
  - lr_scheduler_type: linear
64
  - lr_scheduler_warmup_ratio: 0.1
65
  - num_epochs: 1
 
68
 
69
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
70
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
71
+ | 0.1279 | 1.0 | 516 | 0.0568 | 0.9799 |
72
 
73
 
74
  ### Framework versions
75
 
76
+ - Transformers 4.46.2
77
+ - Pytorch 2.5.1+cu121
78
  - Datasets 3.1.0
79
+ - Tokenizers 0.20.3