
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->

# segformer-b0_DsApollo_scratch

This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
It achieves the following results on the evaluation set:
- Loss: 0.0200
- Mean Iou: 0.7739
- Mean Accuracy: 0.9438
- Overall Accuracy: 0.9904
- Accuracy Background: nan
- Accuracy Haz: 0.9940
- Accuracy Matrix: 0.9861
- Accuracy Porosity: 0.7770
- Accuracy Carbides: 0.9699
- Accuracy Substrate: 0.9922
- Iou Background: 0.0
- Iou Haz: 0.9854
- Iou Matrix: 0.9715
- Iou Porosity: 0.7548
- Iou Carbides: 0.9448
- Iou Substrate: 0.9870

## Model description

More information needed

## Intended uses & limitations

More information needed

## Training and evaluation data

More information needed

## Training procedure

### Training hyperparameters

The following hyperparameters were used during training:
- learning_rate: 6e-05
- train_batch_size: 4
- eval_batch_size: 4
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 10

### Training results

| Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Haz | Accuracy Matrix | Accuracy Porosity | Accuracy Carbides | Accuracy Substrate | Iou Background | Iou Haz | Iou Matrix | Iou Porosity | Iou Carbides | Iou Substrate |
|:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:------------:|:---------------:|:-----------------:|:-----------------:|:------------------:|:--------------:|:-------:|:----------:|:------------:|:------------:|:-------------:|
| 0.2391        | 1.0   | 591  | 0.0839          | 0.6223   | 0.7707        | 0.9700           | nan                 | 0.9892       | 0.9806          | 0.0               | 0.9253            | 0.9583             | 0.0            | 0.9469  | 0.9441     | 0.0          | 0.8911       | 0.9517        |
| 0.1072        | 2.0   | 1182 | 0.1237          | 0.6204   | 0.7713        | 0.9612           | nan                 | 0.9243       | 0.9772          | 0.0               | 0.9610            | 0.9941             | 0.0            | 0.9182  | 0.9570     | 0.0          | 0.9180       | 0.9291        |
| 0.1154        | 3.0   | 1773 | 0.0323          | 0.6429   | 0.7855        | 0.9846           | nan                 | 0.9918       | 0.9840          | 0.0116            | 0.9538            | 0.9861             | 0.0            | 0.9773  | 0.9618     | 0.0116       | 0.9263       | 0.9806        |
| 0.0234        | 4.0   | 2364 | 0.0268          | 0.7248   | 0.8821        | 0.9877           | nan                 | 0.9897       | 0.9889          | 0.4861            | 0.9534            | 0.9923             | 0.0            | 0.9813  | 0.9650     | 0.4855       | 0.9337       | 0.9835        |
| 0.1039        | 5.0   | 2955 | 0.0241          | 0.7508   | 0.9158        | 0.9885           | nan                 | 0.9898       | 0.9797          | 0.6387            | 0.9778            | 0.9929             | 0.0            | 0.9820  | 0.9676     | 0.6328       | 0.9385       | 0.9840        |
| 1.1256        | 6.0   | 3546 | 0.0231          | 0.7622   | 0.9308        | 0.9888           | nan                 | 0.9897       | 0.9811          | 0.7155            | 0.9742            | 0.9935             | 0.0            | 0.9830  | 0.9671     | 0.6971       | 0.9405       | 0.9853        |
| 0.6551        | 7.0   | 4137 | 0.0228          | 0.7712   | 0.9432        | 0.9894           | nan                 | 0.9945       | 0.9879          | 0.7786            | 0.9655            | 0.9894             | 0.0            | 0.9832  | 0.9711     | 0.7445       | 0.9433       | 0.9849        |
| 0.0313        | 8.0   | 4728 | 0.0213          | 0.7759   | 0.9524        | 0.9894           | nan                 | 0.9936       | 0.9855          | 0.8266            | 0.9654            | 0.9908             | 0.0            | 0.9838  | 0.9696     | 0.7731       | 0.9431       | 0.9857        |
| 0.0148        | 9.0   | 5319 | 0.0227          | 0.7695   | 0.9388        | 0.9891           | nan                 | 0.9963       | 0.9859          | 0.7552            | 0.9696            | 0.9871             | 0.0            | 0.9826  | 0.9711     | 0.7347       | 0.9443       | 0.9844        |
| 0.0152        | 10.0  | 5910 | 0.0200          | 0.7739   | 0.9438        | 0.9904           | nan                 | 0.9940       | 0.9861          | 0.7770            | 0.9699            | 0.9922             | 0.0            | 0.9854  | 0.9715     | 0.7548       | 0.9448       | 0.9870        |


### Framework versions

- Transformers 4.35.2
- Pytorch 2.1.0+cu118
- Datasets 2.15.0
- Tokenizers 0.15.0

{"license": "other", "tags": ["generated_from_trainer"], "base_model": "nvidia/mit-b0", "model-index": [{"name": "segformer-b0_DsApollo_scratch", "results": []}]}
transformers tensorboard safetensors segformer generated_from_trainer base_model:nvidia/mit-b0 license:other endpoints_compatible region:us