ihsansatriawan commited on
Commit
20193f4
·
1 Parent(s): 9593bbb

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -31,7 +31,7 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
33
  It achieves the following results on the evaluation set:
34
- - Loss: 2.0995
35
  - Accuracy: 0.1313
36
 
37
  ## Model description
@@ -51,7 +51,7 @@ More information needed
51
  ### Training hyperparameters
52
 
53
  The following hyperparameters were used during training:
54
- - learning_rate: 5e-08
55
  - train_batch_size: 16
56
  - eval_batch_size: 16
57
  - seed: 42
@@ -63,9 +63,9 @@ The following hyperparameters were used during training:
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
- | No log | 1.0 | 40 | 2.0995 | 0.1313 |
67
- | No log | 2.0 | 80 | 2.0994 | 0.1313 |
68
- | No log | 3.0 | 120 | 2.0995 | 0.1313 |
69
 
70
 
71
  ### Framework versions
 
31
 
32
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
33
  It achieves the following results on the evaluation set:
34
+ - Loss: 2.0859
35
  - Accuracy: 0.1313
36
 
37
  ## Model description
 
51
  ### Training hyperparameters
52
 
53
  The following hyperparameters were used during training:
54
+ - learning_rate: 5e-05
55
  - train_batch_size: 16
56
  - eval_batch_size: 16
57
  - seed: 42
 
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
+ | No log | 1.0 | 40 | 2.0859 | 0.1313 |
67
+ | No log | 2.0 | 80 | 2.0857 | 0.1313 |
68
+ | No log | 3.0 | 120 | 2.0854 | 0.1313 |
69
 
70
 
71
  ### Framework versions