update model card README.md

README.md

@@ -14,12 +14,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [ydmeira/segformer-b0-finetuned-pokemon](https://huggingface.co/ydmeira/segformer-b0-finetuned-pokemon) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0225
-- Mean Accuracy: 0.9927
-- Mean Iou: 0.4964
-- Overall Accuracy: 0.9927
-- Per Category Accuracy: [nan, 0.9927247002783977]
-- Per Category Iou: [0.0, 0.9927247002783977]
+- Loss: 0.0190
+- Mean Iou: 0.4972
+- Mean Accuracy: 0.9944
+- Overall Accuracy: 0.9944
+- Per Category Iou: [0.0, 0.9944194724313037]
+- Per Category Accuracy: [nan, 0.9944194724313037]
 
 ## Model description
 
@@ -44,7 +44,7 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 20
+- num_epochs: 50
 
 ### Training results
 
@@ -56,6 +56,12 @@ The following hyperparameters were used during training:
 | 0.0202        | 18.0  | 522  | 0.0226          | 0.9929        | 0.4964   | 0.9929           | [nan, 0.9928635048309444] | [0.0, 0.9928635048309444] |
 | 0.021         | 19.0  | 551  | 0.0226          | 0.9924        | 0.4962   | 0.9924           | [nan, 0.9924163192462797] | [0.0, 0.9924163192462797] |
 | 0.0203        | 20.0  | 580  | 0.0225          | 0.9927        | 0.4964   | 0.9927           | [nan, 0.9927247002783977] | [0.0, 0.9927247002783977] |
+| 0.0184        | 45.0  | 1305 | 0.0193          | 0.4974        | 0.9949   | 0.9949           | [0.0, 0.9948637903811919] | [nan, 0.9948637903811919] |
+| 0.0183        | 46.0  | 1334 | 0.0192          | 0.4972        | 0.9944   | 0.9944           | [0.0, 0.9943721564574917] | [nan, 0.9943721564574917] |
+| 0.0183        | 47.0  | 1363 | 0.0191          | 0.4972        | 0.9945   | 0.9945           | [0.0, 0.994473240583363]  | [nan, 0.994473240583363]  |
+| 0.02          | 48.0  | 1392 | 0.0188          | 0.4972        | 0.9944   | 0.9944           | [0.0, 0.9943908512303614] | [nan, 0.9943908512303614] |
+| 0.0176        | 49.0  | 1421 | 0.0188          | 0.4971        | 0.9942   | 0.9942           | [0.0, 0.9941838300582279] | [nan, 0.9941838300582279] |
+| 0.0189        | 50.0  | 1450 | 0.0190          | 0.4972        | 0.9944   | 0.9944           | [0.0, 0.9944194724313037] | [nan, 0.9944194724313037] |
 
 
 ### Framework versions