End of training

README.md

@@ -0,0 +1,60 @@
+---
+license: other
+base_model: nvidia/mit-b0
+tags:
+- generated_from_trainer
+model-index:
+- name: image_segmentation_text_v2
+  results: []
+---
+
+<!-- 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. -->
+
+# image_segmentation_text_v2
+
+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:
+- eval_loss: 0.0019
+- eval_mean_iou: 1.0
+- eval_mean_accuracy: 1.0
+- eval_overall_accuracy: 1.0
+- eval_per_category_iou: [1.0]
+- eval_per_category_accuracy: [1.0]
+- eval_runtime: 59.0904
+- eval_samples_per_second: 0.846
+- eval_steps_per_second: 0.423
+- epoch: 2.0
+- step: 100
+
+## 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: 2
+- eval_batch_size: 2
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 50
+
+### Framework versions
+
+- Transformers 4.35.2
+- Pytorch 2.1.0+cu121
+- Datasets 2.15.0
+- Tokenizers 0.15.0

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a5d978f77a26bdb12788f12c936e697ec84c3061c5e4210df8183b015363c494
+oid sha256:0ae039c01890f93992bd4b7b8273e55db2240fc396378c4f6292d34f903bb36d
 size 14883748

runs/Dec20_02-26-45_37b315fcf8c8/events.out.tfevents.1703039206.37b315fcf8c8.804.1

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:982c70f6b149dbd3c90b8f9b3968927cbe01109974b0e84edbe7999355a3d34f
-size 22207
+oid sha256:637ea5cc73726fd8d45afcc05d10a64bef19e1ace16eba784d2a53d1a6c30543
+size 25287