KhangTruong/NWPU-Net

README.md

@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1518
+- Loss: 0.1670
 
 ## Model description
 
@@ -47,9 +47,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 0.3107        | 1.0   | 1575 | 0.3146          |
-| 0.1147        | 2.0   | 3150 | 0.1844          |
-| 0.0310        | 3.0   | 4725 | 0.1518          |
+| 0.3427        | 1.0   | 1575 | 0.3408          |
+| 0.0816        | 2.0   | 3150 | 0.1894          |
+| 0.0593        | 3.0   | 4725 | 0.1670          |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:48dbd339fac26b100ebbc22c7c8694f1af4263d0e91e0d3c989074a441024a72
+oid sha256:3cb37bd780bf03e17750aa49b38d0d38e823c985ec1305bc0a3d7e53b8447f6a
 size 343356252