update model card README.md

README.md

@@ -14,8 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 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.
 It achieves the following results on the evaluation set:
-- Loss: 3.5140
-- Rmse: 1.8746
+- Loss: 1.1685
+- Rmse: 1.0810
 
 ## Model description
 
@@ -34,24 +34,27 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1e-05
+- learning_rate: 0.0002
 - train_batch_size: 16
 - eval_batch_size: 64
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 500
-- num_epochs: 5
+- num_epochs: 8
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Rmse   |
 |:-------------:|:-----:|:----:|:---------------:|:------:|
-| No log        | 1.0   | 134  | 3.5140          | 1.8746 |
-| No log        | 2.0   | 268  | 1.6198          | 1.2727 |
-| No log        | 3.0   | 402  | 0.9386          | 0.9688 |
-| 8.1621        | 4.0   | 536  | 0.8200          | 0.9056 |
-| 8.1621        | 5.0   | 670  | 0.7409          | 0.8607 |
+| No log        | 1.0   | 101  | 1.2101          | 1.1001 |
+| No log        | 2.0   | 202  | 0.7917          | 0.8898 |
+| No log        | 3.0   | 303  | 1.0430          | 1.0213 |
+| No log        | 4.0   | 404  | 0.9290          | 0.9639 |
+| 2.8472        | 5.0   | 505  | 0.6923          | 0.8321 |
+| 2.8472        | 6.0   | 606  | 0.8079          | 0.8989 |
+| 2.8472        | 7.0   | 707  | 0.7159          | 0.8461 |
+| 2.8472        | 8.0   | 808  | 0.6934          | 0.8327 |
 
 
 ### Framework versions