tcvrishank commited on
Commit
6082b00
1 Parent(s): adc5401

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -10
README.md CHANGED
@@ -21,7 +21,7 @@ model-index:
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
- value: 0.5
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,8 +31,8 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50) on the imagefolder dataset.
33
  It achieves the following results on the evaluation set:
34
- - Loss: 1.3479
35
- - Accuracy: 0.5
36
 
37
  ## Model description
38
 
@@ -51,7 +51,7 @@ More information needed
51
  ### Training hyperparameters
52
 
53
  The following hyperparameters were used during training:
54
- - learning_rate: 0.0001
55
  - train_batch_size: 64
56
  - eval_batch_size: 8
57
  - seed: 42
@@ -64,12 +64,12 @@ The following hyperparameters were used during training:
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
- | 1.3845 | 1.67 | 10 | 1.3689 | 0.425 |
68
- | 1.3715 | 3.33 | 20 | 1.3776 | 0.35 |
69
- | 1.357 | 5.0 | 30 | 1.3551 | 0.45 |
70
- | 1.3493 | 6.67 | 40 | 1.3612 | 0.425 |
71
- | 1.3377 | 8.33 | 50 | 1.3530 | 0.525 |
72
- | 1.336 | 10.0 | 60 | 1.3479 | 0.5 |
73
 
74
 
75
  ### Framework versions
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.375
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
 
32
  This model is a fine-tuned version of [microsoft/resnet-50](https://huggingface.co/microsoft/resnet-50) on the imagefolder dataset.
33
  It achieves the following results on the evaluation set:
34
+ - Loss: 1.2257
35
+ - Accuracy: 0.375
36
 
37
  ## Model description
38
 
 
51
  ### Training hyperparameters
52
 
53
  The following hyperparameters were used during training:
54
+ - learning_rate: 0.001
55
  - train_batch_size: 64
56
  - eval_batch_size: 8
57
  - seed: 42
 
64
 
65
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
66
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
67
+ | 1.2218 | 1.67 | 10 | 1.3308 | 0.475 |
68
+ | 1.1647 | 3.33 | 20 | 1.1842 | 0.525 |
69
+ | 1.132 | 5.0 | 30 | 1.1902 | 0.5 |
70
+ | 1.1372 | 6.67 | 40 | 1.1355 | 0.5 |
71
+ | 1.1006 | 8.33 | 50 | 1.0911 | 0.45 |
72
+ | 1.0876 | 10.0 | 60 | 1.2257 | 0.375 |
73
 
74
 
75
  ### Framework versions