ShuaHousetable commited on
Commit
b77abee
1 Parent(s): 84ad1b6

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -14,8 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  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 None dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 1.3012
18
- - Rmse: 1.1407
19
 
20
  ## Model description
21
 
@@ -34,7 +34,7 @@ More information needed
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
- - learning_rate: 0.0002
38
  - train_batch_size: 16
39
  - eval_batch_size: 64
40
  - seed: 42
@@ -47,11 +47,11 @@ The following hyperparameters were used during training:
47
 
48
  | Training Loss | Epoch | Step | Validation Loss | Rmse |
49
  |:-------------:|:-----:|:----:|:---------------:|:------:|
50
- | No log | 1.0 | 134 | 0.9896 | 0.9948 |
51
- | No log | 2.0 | 268 | 0.8492 | 0.9215 |
52
- | No log | 3.0 | 402 | 1.3012 | 1.1407 |
53
- | 3.0767 | 4.0 | 536 | 0.8029 | 0.8960 |
54
- | 3.0767 | 5.0 | 670 | 0.5301 | 0.7281 |
55
 
56
 
57
  ### Framework versions
 
14
 
15
  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 None dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Loss: 3.5140
18
+ - Rmse: 1.8746
19
 
20
  ## Model description
21
 
 
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
+ - learning_rate: 1e-05
38
  - train_batch_size: 16
39
  - eval_batch_size: 64
40
  - seed: 42
 
47
 
48
  | Training Loss | Epoch | Step | Validation Loss | Rmse |
49
  |:-------------:|:-----:|:----:|:---------------:|:------:|
50
+ | No log | 1.0 | 134 | 3.5140 | 1.8746 |
51
+ | No log | 2.0 | 268 | 1.6198 | 1.2727 |
52
+ | No log | 3.0 | 402 | 0.9386 | 0.9688 |
53
+ | 8.1621 | 4.0 | 536 | 0.8200 | 0.9056 |
54
+ | 8.1621 | 5.0 | 670 | 0.7409 | 0.8607 |
55
 
56
 
57
  ### Framework versions