End of training
Browse files- README.md +132 -1
- config.json +82 -0
- model.safetensors +3 -0
- preprocessor_config.json +39 -0
- runs/May08_16-16-07_DESKTOP-15M447P/events.out.tfevents.1715199374.DESKTOP-15M447P.37300.3 +3 -0
- runs/May08_16-30-36_DESKTOP-15M447P/events.out.tfevents.1715200246.DESKTOP-15M447P.13804.0 +3 -0
- runs/May08_16-30-36_DESKTOP-15M447P/events.out.tfevents.1715200877.DESKTOP-15M447P.13804.1 +3 -0
- training_args.bin +3 -0
README.md
CHANGED
@@ -1,3 +1,134 @@
|
|
1 |
---
|
2 |
-
license:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
license: other
|
3 |
+
base_model: nvidia/mit-b0
|
4 |
+
tags:
|
5 |
+
- vision
|
6 |
+
- image-segmentation
|
7 |
+
- generated_from_trainer
|
8 |
+
model-index:
|
9 |
+
- name: dungeon-maps-seg-v0.0.1
|
10 |
+
results: []
|
11 |
---
|
12 |
+
|
13 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
14 |
+
should probably proofread and complete it, then remove this comment. -->
|
15 |
+
|
16 |
+
# dungeon-maps-seg-v0.0.1
|
17 |
+
|
18 |
+
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the cephelos/dungeon-maps-seg dataset.
|
19 |
+
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 0.0361
|
21 |
+
- Mean Iou: 0.9518
|
22 |
+
- Mean Accuracy: 0.9783
|
23 |
+
- Overall Accuracy: 0.9893
|
24 |
+
- Accuracy Unlabeled: nan
|
25 |
+
- Accuracy Room: 0.9923
|
26 |
+
- Accuracy Wall: 0.9490
|
27 |
+
- Accuracy Outside: 0.9935
|
28 |
+
- Iou Unlabeled: nan
|
29 |
+
- Iou Room: 0.9857
|
30 |
+
- Iou Wall: 0.8788
|
31 |
+
- Iou Outside: 0.9911
|
32 |
+
|
33 |
+
## Model description
|
34 |
+
|
35 |
+
More information needed
|
36 |
+
|
37 |
+
## Intended uses & limitations
|
38 |
+
|
39 |
+
More information needed
|
40 |
+
|
41 |
+
## Training and evaluation data
|
42 |
+
|
43 |
+
More information needed
|
44 |
+
|
45 |
+
## Training procedure
|
46 |
+
|
47 |
+
### Training hyperparameters
|
48 |
+
|
49 |
+
The following hyperparameters were used during training:
|
50 |
+
- learning_rate: 6e-05
|
51 |
+
- train_batch_size: 2
|
52 |
+
- eval_batch_size: 2
|
53 |
+
- seed: 42
|
54 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
55 |
+
- lr_scheduler_type: linear
|
56 |
+
- num_epochs: 50
|
57 |
+
|
58 |
+
### Training results
|
59 |
+
|
60 |
+
| Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Unlabeled | Accuracy Room | Accuracy Wall | Accuracy Outside | Iou Unlabeled | Iou Room | Iou Wall | Iou Outside |
|
61 |
+
|:-------------:|:-------:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:------------------:|:-------------:|:-------------:|:----------------:|:-------------:|:--------:|:--------:|:-----------:|
|
62 |
+
| 0.2922 | 0.7692 | 20 | 0.2745 | 0.8581 | 0.9561 | 0.9598 | nan | 0.9526 | 0.9466 | 0.9690 | nan | 0.9466 | 0.6646 | 0.9632 |
|
63 |
+
| 0.2099 | 1.5385 | 40 | 0.2072 | 0.8639 | 0.9584 | 0.9625 | nan | 0.9680 | 0.9472 | 0.9599 | nan | 0.9600 | 0.6732 | 0.9584 |
|
64 |
+
| 0.2009 | 2.3077 | 60 | 0.1688 | 0.8968 | 0.9623 | 0.9741 | nan | 0.9718 | 0.9316 | 0.9835 | nan | 0.9649 | 0.7477 | 0.9778 |
|
65 |
+
| 0.1258 | 3.0769 | 80 | 0.1482 | 0.8991 | 0.9676 | 0.9745 | nan | 0.9773 | 0.9492 | 0.9762 | nan | 0.9708 | 0.7529 | 0.9736 |
|
66 |
+
| 0.1624 | 3.8462 | 100 | 0.1333 | 0.9115 | 0.9682 | 0.9785 | nan | 0.9807 | 0.9410 | 0.9829 | nan | 0.9734 | 0.7817 | 0.9795 |
|
67 |
+
| 0.1098 | 4.6154 | 120 | 0.1079 | 0.9173 | 0.9624 | 0.9805 | nan | 0.9859 | 0.9145 | 0.9868 | nan | 0.9753 | 0.7950 | 0.9817 |
|
68 |
+
| 0.1629 | 5.3846 | 140 | 0.1041 | 0.9195 | 0.9711 | 0.9806 | nan | 0.9790 | 0.9462 | 0.9881 | nan | 0.9738 | 0.8013 | 0.9833 |
|
69 |
+
| 0.1243 | 6.1538 | 160 | 0.0872 | 0.9243 | 0.9675 | 0.9821 | nan | 0.9852 | 0.9288 | 0.9884 | nan | 0.9766 | 0.8125 | 0.9836 |
|
70 |
+
| 0.0974 | 6.9231 | 180 | 0.0996 | 0.9217 | 0.9731 | 0.9811 | nan | 0.9754 | 0.9525 | 0.9915 | nan | 0.9717 | 0.8073 | 0.9861 |
|
71 |
+
| 0.0861 | 7.6923 | 200 | 0.0798 | 0.9248 | 0.9706 | 0.9821 | nan | 0.9829 | 0.9403 | 0.9886 | nan | 0.9764 | 0.8142 | 0.9836 |
|
72 |
+
| 0.0928 | 8.4615 | 220 | 0.0718 | 0.9276 | 0.9740 | 0.9828 | nan | 0.9830 | 0.9507 | 0.9882 | nan | 0.9773 | 0.8209 | 0.9847 |
|
73 |
+
| 0.0583 | 9.2308 | 240 | 0.0726 | 0.9240 | 0.9686 | 0.9822 | nan | 0.9870 | 0.9326 | 0.9862 | nan | 0.9789 | 0.8111 | 0.9821 |
|
74 |
+
| 0.0886 | 10.0 | 260 | 0.0700 | 0.9296 | 0.9740 | 0.9835 | nan | 0.9845 | 0.9491 | 0.9885 | nan | 0.9786 | 0.8250 | 0.9852 |
|
75 |
+
| 0.1133 | 10.7692 | 280 | 0.0651 | 0.9322 | 0.9633 | 0.9848 | nan | 0.9912 | 0.9064 | 0.9922 | nan | 0.9794 | 0.8301 | 0.9872 |
|
76 |
+
| 0.0821 | 11.5385 | 300 | 0.0616 | 0.9302 | 0.9721 | 0.9836 | nan | 0.9833 | 0.9417 | 0.9912 | nan | 0.9779 | 0.8270 | 0.9857 |
|
77 |
+
| 0.07 | 12.3077 | 320 | 0.0586 | 0.9394 | 0.9690 | 0.9864 | nan | 0.9896 | 0.9232 | 0.9942 | nan | 0.9810 | 0.8485 | 0.9887 |
|
78 |
+
| 0.076 | 13.0769 | 340 | 0.0566 | 0.9349 | 0.9651 | 0.9854 | nan | 0.9919 | 0.9113 | 0.9920 | nan | 0.9803 | 0.8365 | 0.9878 |
|
79 |
+
| 0.0577 | 13.8462 | 360 | 0.0570 | 0.9378 | 0.9755 | 0.9857 | nan | 0.9850 | 0.9488 | 0.9926 | nan | 0.9797 | 0.8452 | 0.9886 |
|
80 |
+
| 0.1261 | 14.6154 | 380 | 0.0548 | 0.9403 | 0.9739 | 0.9864 | nan | 0.9867 | 0.9410 | 0.9939 | nan | 0.9808 | 0.8511 | 0.9891 |
|
81 |
+
| 0.0583 | 15.3846 | 400 | 0.0523 | 0.9428 | 0.9736 | 0.9871 | nan | 0.9895 | 0.9379 | 0.9934 | nan | 0.9820 | 0.8566 | 0.9896 |
|
82 |
+
| 0.0602 | 16.1538 | 420 | 0.0488 | 0.9409 | 0.9737 | 0.9866 | nan | 0.9899 | 0.9394 | 0.9917 | nan | 0.9820 | 0.8519 | 0.9887 |
|
83 |
+
| 0.0728 | 16.9231 | 440 | 0.0504 | 0.9380 | 0.9716 | 0.9860 | nan | 0.9907 | 0.9335 | 0.9905 | nan | 0.9819 | 0.8448 | 0.9873 |
|
84 |
+
| 0.0507 | 17.6923 | 460 | 0.0503 | 0.9378 | 0.9739 | 0.9858 | nan | 0.9892 | 0.9424 | 0.9901 | nan | 0.9820 | 0.8445 | 0.9869 |
|
85 |
+
| 0.077 | 18.4615 | 480 | 0.0474 | 0.9429 | 0.9740 | 0.9871 | nan | 0.9876 | 0.9396 | 0.9949 | nan | 0.9819 | 0.8570 | 0.9897 |
|
86 |
+
| 0.2137 | 19.2308 | 500 | 0.0500 | 0.9413 | 0.9763 | 0.9866 | nan | 0.9892 | 0.9489 | 0.9907 | nan | 0.9823 | 0.8532 | 0.9882 |
|
87 |
+
| 0.0991 | 20.0 | 520 | 0.0459 | 0.9440 | 0.9719 | 0.9875 | nan | 0.9899 | 0.9309 | 0.9950 | nan | 0.9827 | 0.8595 | 0.9898 |
|
88 |
+
| 0.0691 | 20.7692 | 540 | 0.0447 | 0.9451 | 0.9743 | 0.9877 | nan | 0.9906 | 0.9390 | 0.9933 | nan | 0.9831 | 0.8623 | 0.9897 |
|
89 |
+
| 0.0602 | 21.5385 | 560 | 0.0447 | 0.9462 | 0.9754 | 0.9879 | nan | 0.9885 | 0.9424 | 0.9952 | nan | 0.9828 | 0.8654 | 0.9904 |
|
90 |
+
| 0.0469 | 22.3077 | 580 | 0.0429 | 0.9466 | 0.9767 | 0.9879 | nan | 0.9889 | 0.9471 | 0.9940 | nan | 0.9830 | 0.8664 | 0.9903 |
|
91 |
+
| 0.0553 | 23.0769 | 600 | 0.0445 | 0.9468 | 0.9722 | 0.9882 | nan | 0.9913 | 0.9301 | 0.9952 | nan | 0.9832 | 0.8666 | 0.9906 |
|
92 |
+
| 0.0671 | 23.8462 | 620 | 0.0424 | 0.9455 | 0.9748 | 0.9878 | nan | 0.9900 | 0.9407 | 0.9938 | nan | 0.9833 | 0.8635 | 0.9898 |
|
93 |
+
| 0.0431 | 24.6154 | 640 | 0.0417 | 0.9475 | 0.9732 | 0.9883 | nan | 0.9921 | 0.9331 | 0.9943 | nan | 0.9836 | 0.8681 | 0.9907 |
|
94 |
+
| 0.0381 | 25.3846 | 660 | 0.0429 | 0.9449 | 0.9763 | 0.9876 | nan | 0.9881 | 0.9467 | 0.9942 | nan | 0.9827 | 0.8620 | 0.9901 |
|
95 |
+
| 0.0503 | 26.1538 | 680 | 0.0403 | 0.9471 | 0.9746 | 0.9882 | nan | 0.9924 | 0.9384 | 0.9929 | nan | 0.9841 | 0.8669 | 0.9902 |
|
96 |
+
| 0.0685 | 26.9231 | 700 | 0.0410 | 0.9496 | 0.9743 | 0.9888 | nan | 0.9913 | 0.9361 | 0.9957 | nan | 0.9842 | 0.8732 | 0.9912 |
|
97 |
+
| 0.0381 | 27.6923 | 720 | 0.0398 | 0.9494 | 0.9771 | 0.9887 | nan | 0.9906 | 0.9466 | 0.9942 | nan | 0.9843 | 0.8729 | 0.9909 |
|
98 |
+
| 0.0587 | 28.4615 | 740 | 0.0397 | 0.9500 | 0.9760 | 0.9889 | nan | 0.9913 | 0.9421 | 0.9947 | nan | 0.9843 | 0.8743 | 0.9913 |
|
99 |
+
| 0.0573 | 29.2308 | 760 | 0.0402 | 0.9489 | 0.9756 | 0.9887 | nan | 0.9913 | 0.9411 | 0.9945 | nan | 0.9845 | 0.8715 | 0.9908 |
|
100 |
+
| 0.0686 | 30.0 | 780 | 0.0386 | 0.9499 | 0.9763 | 0.9889 | nan | 0.9914 | 0.9433 | 0.9944 | nan | 0.9844 | 0.8740 | 0.9912 |
|
101 |
+
| 0.037 | 30.7692 | 800 | 0.0386 | 0.9503 | 0.9752 | 0.9890 | nan | 0.9925 | 0.9387 | 0.9944 | nan | 0.9849 | 0.8748 | 0.9911 |
|
102 |
+
| 0.0565 | 31.5385 | 820 | 0.0389 | 0.9497 | 0.9773 | 0.9888 | nan | 0.9898 | 0.9471 | 0.9950 | nan | 0.9840 | 0.8738 | 0.9913 |
|
103 |
+
| 0.0405 | 32.3077 | 840 | 0.0383 | 0.9483 | 0.9743 | 0.9886 | nan | 0.9933 | 0.9366 | 0.9930 | nan | 0.9848 | 0.8698 | 0.9903 |
|
104 |
+
| 0.0618 | 33.0769 | 860 | 0.0383 | 0.9497 | 0.9757 | 0.9889 | nan | 0.9920 | 0.9408 | 0.9942 | nan | 0.9847 | 0.8734 | 0.9910 |
|
105 |
+
| 0.0398 | 33.8462 | 880 | 0.0379 | 0.9494 | 0.9766 | 0.9888 | nan | 0.9917 | 0.9446 | 0.9936 | nan | 0.9846 | 0.8729 | 0.9908 |
|
106 |
+
| 0.0488 | 34.6154 | 900 | 0.0376 | 0.9501 | 0.9769 | 0.9889 | nan | 0.9915 | 0.9450 | 0.9941 | nan | 0.9851 | 0.8745 | 0.9907 |
|
107 |
+
| 0.0574 | 35.3846 | 920 | 0.0379 | 0.9512 | 0.9762 | 0.9892 | nan | 0.9914 | 0.9419 | 0.9953 | nan | 0.9849 | 0.8773 | 0.9914 |
|
108 |
+
| 0.0331 | 36.1538 | 940 | 0.0368 | 0.9514 | 0.9764 | 0.9893 | nan | 0.9921 | 0.9424 | 0.9947 | nan | 0.9852 | 0.8777 | 0.9913 |
|
109 |
+
| 0.0578 | 36.9231 | 960 | 0.0368 | 0.9520 | 0.9770 | 0.9894 | nan | 0.9916 | 0.9443 | 0.9951 | nan | 0.9852 | 0.8790 | 0.9917 |
|
110 |
+
| 0.0471 | 37.6923 | 980 | 0.0369 | 0.9517 | 0.9779 | 0.9893 | nan | 0.9912 | 0.9480 | 0.9947 | nan | 0.9852 | 0.8786 | 0.9915 |
|
111 |
+
| 0.0388 | 38.4615 | 1000 | 0.0369 | 0.9511 | 0.9776 | 0.9892 | nan | 0.9904 | 0.9473 | 0.9952 | nan | 0.9846 | 0.8770 | 0.9916 |
|
112 |
+
| 0.0455 | 39.2308 | 1020 | 0.0367 | 0.9517 | 0.9753 | 0.9894 | nan | 0.9928 | 0.9379 | 0.9950 | nan | 0.9853 | 0.8784 | 0.9915 |
|
113 |
+
| 0.0359 | 40.0 | 1040 | 0.0360 | 0.9516 | 0.9773 | 0.9893 | nan | 0.9917 | 0.9457 | 0.9945 | nan | 0.9853 | 0.8783 | 0.9913 |
|
114 |
+
| 0.0281 | 40.7692 | 1060 | 0.0363 | 0.9519 | 0.9775 | 0.9894 | nan | 0.9917 | 0.9462 | 0.9946 | nan | 0.9854 | 0.8790 | 0.9913 |
|
115 |
+
| 0.0394 | 41.5385 | 1080 | 0.0367 | 0.9508 | 0.9769 | 0.9891 | nan | 0.9922 | 0.9446 | 0.9939 | nan | 0.9854 | 0.8761 | 0.9909 |
|
116 |
+
| 0.0286 | 42.3077 | 1100 | 0.0360 | 0.9525 | 0.9761 | 0.9896 | nan | 0.9924 | 0.9405 | 0.9953 | nan | 0.9855 | 0.8804 | 0.9917 |
|
117 |
+
| 0.028 | 43.0769 | 1120 | 0.0363 | 0.9509 | 0.9791 | 0.9891 | nan | 0.9909 | 0.9530 | 0.9936 | nan | 0.9850 | 0.8767 | 0.9911 |
|
118 |
+
| 0.0523 | 43.8462 | 1140 | 0.0366 | 0.9526 | 0.9777 | 0.9895 | nan | 0.9919 | 0.9466 | 0.9947 | nan | 0.9856 | 0.8806 | 0.9915 |
|
119 |
+
| 0.0492 | 44.6154 | 1160 | 0.0364 | 0.9523 | 0.9764 | 0.9895 | nan | 0.9926 | 0.9419 | 0.9948 | nan | 0.9856 | 0.8799 | 0.9915 |
|
120 |
+
| 0.0331 | 45.3846 | 1180 | 0.0356 | 0.9523 | 0.9781 | 0.9894 | nan | 0.9906 | 0.9484 | 0.9954 | nan | 0.9852 | 0.8799 | 0.9917 |
|
121 |
+
| 0.0443 | 46.1538 | 1200 | 0.0358 | 0.9533 | 0.9772 | 0.9897 | nan | 0.9921 | 0.9443 | 0.9953 | nan | 0.9857 | 0.8824 | 0.9918 |
|
122 |
+
| 0.0331 | 46.9231 | 1220 | 0.0356 | 0.9527 | 0.9771 | 0.9896 | nan | 0.9929 | 0.9441 | 0.9943 | nan | 0.9858 | 0.8808 | 0.9915 |
|
123 |
+
| 0.0546 | 47.6923 | 1240 | 0.0357 | 0.9532 | 0.9774 | 0.9897 | nan | 0.9916 | 0.9450 | 0.9956 | nan | 0.9856 | 0.8821 | 0.9919 |
|
124 |
+
| 0.0297 | 48.4615 | 1260 | 0.0351 | 0.9526 | 0.9776 | 0.9896 | nan | 0.9925 | 0.9461 | 0.9942 | nan | 0.9857 | 0.8807 | 0.9915 |
|
125 |
+
| 0.053 | 49.2308 | 1280 | 0.0349 | 0.9527 | 0.9779 | 0.9896 | nan | 0.9921 | 0.9471 | 0.9945 | nan | 0.9856 | 0.8809 | 0.9916 |
|
126 |
+
| 0.0474 | 50.0 | 1300 | 0.0361 | 0.9518 | 0.9783 | 0.9893 | nan | 0.9923 | 0.9490 | 0.9935 | nan | 0.9857 | 0.8788 | 0.9911 |
|
127 |
+
|
128 |
+
|
129 |
+
### Framework versions
|
130 |
+
|
131 |
+
- Transformers 4.40.2
|
132 |
+
- Pytorch 2.2.0+cpu
|
133 |
+
- Datasets 2.19.1
|
134 |
+
- Tokenizers 0.19.1
|
config.json
ADDED
@@ -0,0 +1,82 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "nvidia/mit-b0",
|
3 |
+
"architectures": [
|
4 |
+
"SegformerForSemanticSegmentation"
|
5 |
+
],
|
6 |
+
"attention_probs_dropout_prob": 0.0,
|
7 |
+
"classifier_dropout_prob": 0.1,
|
8 |
+
"decoder_hidden_size": 256,
|
9 |
+
"depths": [
|
10 |
+
2,
|
11 |
+
2,
|
12 |
+
2,
|
13 |
+
2
|
14 |
+
],
|
15 |
+
"downsampling_rates": [
|
16 |
+
1,
|
17 |
+
4,
|
18 |
+
8,
|
19 |
+
16
|
20 |
+
],
|
21 |
+
"drop_path_rate": 0.1,
|
22 |
+
"hidden_act": "gelu",
|
23 |
+
"hidden_dropout_prob": 0.0,
|
24 |
+
"hidden_sizes": [
|
25 |
+
32,
|
26 |
+
64,
|
27 |
+
160,
|
28 |
+
256
|
29 |
+
],
|
30 |
+
"id2label": {
|
31 |
+
"0": "unlabeled",
|
32 |
+
"1": "room",
|
33 |
+
"2": "wall",
|
34 |
+
"3": "outside"
|
35 |
+
},
|
36 |
+
"image_size": 224,
|
37 |
+
"initializer_range": 0.02,
|
38 |
+
"label2id": {
|
39 |
+
"outside": 3,
|
40 |
+
"room": 1,
|
41 |
+
"unlabeled": 0,
|
42 |
+
"wall": 2
|
43 |
+
},
|
44 |
+
"layer_norm_eps": 1e-06,
|
45 |
+
"mlp_ratios": [
|
46 |
+
4,
|
47 |
+
4,
|
48 |
+
4,
|
49 |
+
4
|
50 |
+
],
|
51 |
+
"model_type": "segformer",
|
52 |
+
"num_attention_heads": [
|
53 |
+
1,
|
54 |
+
2,
|
55 |
+
5,
|
56 |
+
8
|
57 |
+
],
|
58 |
+
"num_channels": 3,
|
59 |
+
"num_encoder_blocks": 4,
|
60 |
+
"patch_sizes": [
|
61 |
+
7,
|
62 |
+
3,
|
63 |
+
3,
|
64 |
+
3
|
65 |
+
],
|
66 |
+
"reshape_last_stage": true,
|
67 |
+
"semantic_loss_ignore_index": 255,
|
68 |
+
"sr_ratios": [
|
69 |
+
8,
|
70 |
+
4,
|
71 |
+
2,
|
72 |
+
1
|
73 |
+
],
|
74 |
+
"strides": [
|
75 |
+
4,
|
76 |
+
2,
|
77 |
+
2,
|
78 |
+
2
|
79 |
+
],
|
80 |
+
"torch_dtype": "float32",
|
81 |
+
"transformers_version": "4.40.2"
|
82 |
+
}
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2a5d477657072c595eb2dc5753b405afbae638d6ba57fd77934d395e5d0d4bfa
|
3 |
+
size 14886832
|
preprocessor_config.json
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_valid_processor_keys": [
|
3 |
+
"images",
|
4 |
+
"segmentation_maps",
|
5 |
+
"do_resize",
|
6 |
+
"size",
|
7 |
+
"resample",
|
8 |
+
"do_rescale",
|
9 |
+
"rescale_factor",
|
10 |
+
"do_normalize",
|
11 |
+
"image_mean",
|
12 |
+
"image_std",
|
13 |
+
"do_reduce_labels",
|
14 |
+
"return_tensors",
|
15 |
+
"data_format",
|
16 |
+
"input_data_format"
|
17 |
+
],
|
18 |
+
"do_normalize": true,
|
19 |
+
"do_reduce_labels": false,
|
20 |
+
"do_rescale": true,
|
21 |
+
"do_resize": true,
|
22 |
+
"image_mean": [
|
23 |
+
0.485,
|
24 |
+
0.456,
|
25 |
+
0.406
|
26 |
+
],
|
27 |
+
"image_processor_type": "SegformerFeatureExtractor",
|
28 |
+
"image_std": [
|
29 |
+
0.229,
|
30 |
+
0.224,
|
31 |
+
0.225
|
32 |
+
],
|
33 |
+
"resample": 2,
|
34 |
+
"rescale_factor": 0.00392156862745098,
|
35 |
+
"size": {
|
36 |
+
"height": 512,
|
37 |
+
"width": 512
|
38 |
+
}
|
39 |
+
}
|
runs/May08_16-16-07_DESKTOP-15M447P/events.out.tfevents.1715199374.DESKTOP-15M447P.37300.3
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4d05733870d1c8a1011d2757222f48092aabac9cbb2b778d19a2d7c4a9dd286c
|
3 |
+
size 5245
|
runs/May08_16-30-36_DESKTOP-15M447P/events.out.tfevents.1715200246.DESKTOP-15M447P.13804.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7d8460bce5d8ed99f18976e9913a2758735946501e196d087f1148a0467e2211
|
3 |
+
size 27979
|
runs/May08_16-30-36_DESKTOP-15M447P/events.out.tfevents.1715200877.DESKTOP-15M447P.13804.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e74eb8198014487dbd48dc0709599304c7895a94e796b82c8d4fbcfa5b9a851a
|
3 |
+
size 337275
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b377b43a56b8d63ef9782f8c8c82a407ffea5c58c6f031b01119549859b2c240
|
3 |
+
size 5048
|