End of training

README.md

@@ -17,6 +17,27 @@ should probably proofread and complete it, then remove this comment. -->
 # earthquake
 
 This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the gokceKy/earthquake dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.6395
+- Mean Iou: 0.1763
+- Mean Accuracy: 0.4255
+- Overall Accuracy: 0.6211
+- Accuracy Background: nan
+- Accuracy Car: nan
+- Accuracy Earthquake-roads: 0.4591
+- Accuracy Other: 0.8175
+- Accuracy Road: 0.0
+- Accuracy Road-cracks: nan
+- Accuracy Sky: nan
+- Accuracy Wall: nan
+- Iou Background: 0.0
+- Iou Car: 0.0
+- Iou Earthquake-roads: 0.4499
+- Iou Other: 0.7841
+- Iou Road: 0.0
+- Iou Road-cracks: 0.0
+- Iou Sky: 0.0
+- Iou Wall: nan
 
 ## Model description
 
@@ -41,10 +62,14 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 2
+- num_epochs: 5
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Car | Accuracy Earthquake-roads | Accuracy Other | Accuracy Road | Accuracy Road-cracks | Accuracy Sky | Accuracy Wall | Iou Background | Iou Car | Iou Earthquake-roads | Iou Other | Iou Road | Iou Road-cracks | Iou Sky | Iou Wall |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:------------:|:-------------------------:|:--------------:|:-------------:|:--------------------:|:------------:|:-------------:|:--------------:|:-------:|:--------------------:|:---------:|:--------:|:---------------:|:-------:|:--------:|
+| 1.0736        | 2.5   | 5    | 1.6917          | 0.1818   | 0.3763        | 0.5984           | nan                 | nan          | 0.3113                    | 0.8176         | 0.0           | nan                  | nan          | nan           | 0.0            | 0.0     | 0.3077               | 0.7830    | 0.0      | 0.0             | nan     | nan      |
+| 0.8949        | 5.0   | 10   | 1.6395          | 0.1763   | 0.4255        | 0.6211           | nan                 | nan          | 0.4591                    | 0.8175         | 0.0           | nan                  | nan          | nan           | 0.0            | 0.0     | 0.4499               | 0.7841    | 0.0      | 0.0             | 0.0     | nan      |
 
 
 ### Framework versions

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "nvidia/mit-b0",
+  "_name_or_path": "gokceKy/earthquake",
   "architectures": [
     "SegformerForSemanticSegmentation"
   ],

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:89e5db610c37eca878df3e251906005bd97b18e7224259db119e671281af72ab
+oid sha256:f459bf3e475699dbe7616ffb51ed473c66bd41702fd0bf8974ba5da24e9dc2b4
 size 14890952

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:568b1b28b3171499e3543eff89f500401703736d220fa43b29535ff29e26c83d
+oid sha256:792d5fc0abee56080e2e2acd7cfdd54c8b46488ea6c4b16845dba139f207254b
 size 5304