End of training

README.md

@@ -15,12 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [nvidia/mit-b3](https://huggingface.co/nvidia/mit-b3) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1565
-- Mean Iou: 0.7861
-- Mean Accuracy: 0.8675
-- Overall Accuracy: 0.9459
-- Per Category Iou: [0.9403407357445772, 0.6318399840728189]
-- Per Category Accuracy: [0.9710837035497308, 0.7638247131743141]
+- Loss: 0.1438
+- Mean Iou: 0.7943
+- Mean Accuracy: 0.8621
+- Overall Accuracy: 0.9463
+- Per Category Iou: [0.9404316292856895, 0.6481512759943856]
+- Per Category Accuracy: [0.9766097421311045, 0.7476136089750051]
 
 ## Model description
 
@@ -51,36 +51,36 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step  | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Per Category Iou                         | Per Category Accuracy                    |
 |:-------------:|:-----:|:-----:|:---------------:|:--------:|:-------------:|:----------------:|:----------------------------------------:|:----------------------------------------:|
-| 0.0823        | 1.0   | 514   | 0.1854          | 0.7127   | 0.7910        | 0.9269           | [0.9210383806308944, 0.5042918262988731] | [0.9705715523625104, 0.6114988388304967] |
-| 0.0687        | 2.0   | 1028  | 0.1841          | 0.7393   | 0.8432        | 0.9292           | [0.9223533231277112, 0.5562647361107461] | [0.9568919509142975, 0.729491028137852]  |
-| 0.1894        | 3.0   | 1542  | 0.1778          | 0.7398   | 0.8404        | 0.9300           | [0.9232526693838075, 0.5563397693217451] | [0.958784482352132, 0.7219834134333204]  |
-| 0.7538        | 4.0   | 2056  | 0.1772          | 0.7416   | 0.8359        | 0.9317           | [0.9252002744357394, 0.5579050317848258] | [0.9624271178784354, 0.7093339081402684] |
-| 0.114         | 5.0   | 2570  | 0.1715          | 0.7477   | 0.8323        | 0.9352           | [0.9291644605624902, 0.5663348012087291] | [0.9682212065124535, 0.6963471385392261] |
-| 0.2602        | 6.0   | 3084  | 0.1628          | 0.7522   | 0.8255        | 0.9384           | [0.9328512114347738, 0.5716309109545779] | [0.9746503265081415, 0.6763104945971122] |
-| 0.0449        | 7.0   | 3598  | 0.1646          | 0.7564   | 0.8429        | 0.9372           | [0.9311867725956636, 0.5817017708483792] | [0.9675049740765689, 0.7182516075487344] |
-| 0.3232        | 8.0   | 4112  | 0.1629          | 0.7505   | 0.8172        | 0.9391           | [0.9338111104403589, 0.5670995709520322] | [0.9782553311094724, 0.6561807648822252] |
-| 0.5253        | 9.0   | 4626  | 0.1692          | 0.7591   | 0.8656        | 0.9346           | [0.927839360080974, 0.5902658658166116]  | [0.9568132965794497, 0.7744158284256216] |
-| 0.2831        | 10.0  | 5140  | 0.1696          | 0.7582   | 0.8544        | 0.9361           | [0.929666702131827, 0.5867196051740685]  | [0.9622762425764686, 0.7466089251133196] |
-| 0.1156        | 11.0  | 5654  | 0.1650          | 0.7597   | 0.8538        | 0.9368           | [0.9305677111384715, 0.5889272788978366] | [0.9635390166837846, 0.7440459085851031] |
-| 0.1219        | 12.0  | 6168  | 0.1628          | 0.7691   | 0.8611        | 0.9398           | [0.9336816581722548, 0.6046120174129817] | [0.9650616817423817, 0.7572113671732611] |
-| 0.2278        | 13.0  | 6682  | 0.1595          | 0.7706   | 0.8531        | 0.9416           | [0.9358804294399143, 0.6052538747817555] | [0.9700634208422755, 0.7361460430652129] |
-| 0.0709        | 14.0  | 7196  | 0.1659          | 0.7672   | 0.8542        | 0.9400           | [0.9340658745878584, 0.600268583614642]  | [0.9675667230034556, 0.7409090690667602] |
-| 0.1536        | 15.0  | 7710  | 0.1585          | 0.7738   | 0.8516        | 0.9432           | [0.9376439019013373, 0.6099064919226352] | [0.9726173252892607, 0.7305525217251622] |
-| 0.1594        | 16.0  | 8224  | 0.1596          | 0.7769   | 0.8651        | 0.9425           | [0.9366024648456509, 0.6172773587238504] | [0.9673430850771165, 0.7629001710004867] |
-| 0.0789        | 17.0  | 8738  | 0.1577          | 0.7745   | 0.8499        | 0.9437           | [0.9382865730369421, 0.6106973639701876] | [0.9738904193926676, 0.7258834052642887] |
-| 0.057         | 18.0  | 9252  | 0.1608          | 0.7769   | 0.8643        | 0.9426           | [0.9367437147934611, 0.6171211837012913] | [0.967748178183342, 0.760901229542239]   |
-| 0.2001        | 19.0  | 9766  | 0.1636          | 0.7758   | 0.8654        | 0.9420           | [0.9360252509546998, 0.615599666996662]  | [0.9665703017234261, 0.7642633009340841] |
-| 0.0952        | 20.0  | 10280 | 0.1589          | 0.7774   | 0.8567        | 0.9439           | [0.938335347249839, 0.6164059298692968]  | [0.9719200128462095, 0.741442664724405]  |
-| 0.0547        | 21.0  | 10794 | 0.1624          | 0.7701   | 0.8477        | 0.9422           | [0.9366784658677474, 0.603460893036555]  | [0.9726164795437247, 0.7228356013265409] |
-| 0.2193        | 22.0  | 11308 | 0.1637          | 0.7813   | 0.8705        | 0.9435           | [0.937652191484115, 0.6249915686388726]  | [0.966972715051661, 0.7741064349316032]  |
-| 0.107         | 23.0  | 11822 | 0.1564          | 0.7844   | 0.8654        | 0.9455           | [0.9399404830453398, 0.6288692515144549] | [0.971215620847829, 0.7596341372163922]  |
-| 0.0686        | 24.0  | 12336 | 0.1616          | 0.7814   | 0.8664        | 0.9441           | [0.9384157637282583, 0.6243761952101737] | [0.9691135345722435, 0.7636881736447817] |
-| 0.0755        | 25.0  | 12850 | 0.1600          | 0.7806   | 0.8551        | 0.9454           | [0.9400808454159616, 0.6211992243461064] | [0.9744623524356968, 0.7357995113683403] |
-| 0.0777        | 26.0  | 13364 | 0.1554          | 0.7827   | 0.8617        | 0.9453           | [0.93983539112418, 0.6255737994703556]   | [0.972208915720203, 0.7511646361932037]  |
-| 0.14          | 27.0  | 13878 | 0.1606          | 0.7843   | 0.8689        | 0.9450           | [0.939284077651327, 0.6292689574617442]  | [0.9694035492690203, 0.7683542696233527] |
-| 0.1203        | 28.0  | 14392 | 0.1576          | 0.7868   | 0.8666        | 0.9463           | [0.9408085456144354, 0.632817365056415]  | [0.9718973297608788, 0.7612868215670674] |
-| 0.0344        | 29.0  | 14906 | 0.1543          | 0.7860   | 0.8637        | 0.9463           | [0.9409194486606696, 0.6309827708873241] | [0.9728868425352379, 0.7545694062212325] |
-| 0.0226        | 30.0  | 15420 | 0.1565          | 0.7861   | 0.8675        | 0.9459           | [0.9403407357445772, 0.6318399840728189] | [0.9710837035497308, 0.7638247131743141] |
+| 0.2683        | 1.0   | 514   | 0.1919          | 0.7319   | 0.8203        | 0.9254           | [0.9180806580950376, 0.5456807971830461] | [0.9632417779264837, 0.6772761097220785] |
+| 0.2221        | 2.0   | 1028  | 0.1848          | 0.7247   | 0.7906        | 0.9285           | [0.9222502676042136, 0.5272319457425635] | [0.9780221080448669, 0.6032531950921799] |
+| 0.2051        | 3.0   | 1542  | 0.1798          | 0.7523   | 0.8476        | 0.9301           | [0.9226057238240142, 0.5819673887381934] | [0.9598172268335441, 0.7353887410800054] |
+| 0.1996        | 4.0   | 2056  | 0.1662          | 0.7530   | 0.8151        | 0.9366           | [0.9305887422755489, 0.5753549591140691] | [0.9802446445708194, 0.6499256145141094] |
+| 0.1878        | 5.0   | 2570  | 0.1613          | 0.7695   | 0.8443        | 0.9386           | [0.9321977696827667, 0.6067612442121683] | [0.972541727740175, 0.7160658070940397]  |
+| 0.1848        | 6.0   | 3084  | 0.1579          | 0.7659   | 0.8299        | 0.9395           | [0.9335431699668292, 0.5982432303010027] | [0.9789606327074537, 0.6808199676860336] |
+| 0.1739        | 7.0   | 3598  | 0.1543          | 0.7799   | 0.8516        | 0.9419           | [0.9357233119518715, 0.6240100443072397] | [0.9744037993799354, 0.7287987165292924] |
+| 0.1742        | 8.0   | 4112  | 0.1607          | 0.7737   | 0.8667        | 0.9369           | [0.9296752366973121, 0.6176824708083143] | [0.9620707480253384, 0.7713870499643393] |
+| 0.1631        | 9.0   | 4626  | 0.1553          | 0.7803   | 0.8636        | 0.9402           | [0.9335684843016983, 0.626952054351963]  | [0.9678138198093251, 0.7593407854928076] |
+| 0.1631        | 10.0  | 5140  | 0.1564          | 0.7679   | 0.8406        | 0.9386           | [0.9322118932460649, 0.6035637849191774] | [0.9738013828280326, 0.7073043940670594] |
+| 0.1577        | 11.0  | 5654  | 0.1499          | 0.7836   | 0.8520        | 0.9434           | [0.9373956576514434, 0.6297874494960901] | [0.9763014923228973, 0.7277053858903674] |
+| 0.1522        | 12.0  | 6168  | 0.1515          | 0.7781   | 0.8454        | 0.9422           | [0.9361248378060324, 0.6200926437502748] | [0.9769566337905029, 0.7138379976069665] |
+| 0.1486        | 13.0  | 6682  | 0.1531          | 0.7766   | 0.8485        | 0.9410           | [0.9347900233191081, 0.618411826576312]  | [0.9743083731504567, 0.7226475641364578] |
+| 0.146         | 14.0  | 7196  | 0.1568          | 0.7835   | 0.8667        | 0.9412           | [0.9345511743761084, 0.6325140238903502] | [0.9679584694275379, 0.7654769806378111] |
+| 0.1453        | 15.0  | 7710  | 0.1485          | 0.7837   | 0.8473        | 0.9442           | [0.938357945982732, 0.6290898844632448]  | [0.9790292010707323, 0.7156414504087399] |
+| 0.1439        | 16.0  | 8224  | 0.1492          | 0.7896   | 0.8623        | 0.9444           | [0.9382510027549076, 0.6409378390750833] | [0.973914066583286, 0.7506284726295933]  |
+| 0.1385        | 17.0  | 8738  | 0.1494          | 0.7790   | 0.8369        | 0.9439           | [0.938230866841315, 0.6196920559815114]  | [0.9823649345853678, 0.6913889357135595] |
+| 0.1375        | 18.0  | 9252  | 0.1515          | 0.7847   | 0.8549        | 0.9435           | [0.9373782356302373, 0.6321131725666339] | [0.975314658131637, 0.7344851844673589]  |
+| 0.1353        | 19.0  | 9766  | 0.1450          | 0.7929   | 0.8610        | 0.9459           | [0.9399812074698386, 0.6457742806194801] | [0.9764190916833184, 0.7456795800307741] |
+| 0.1317        | 20.0  | 10280 | 0.1453          | 0.7906   | 0.8584        | 0.9453           | [0.9394006951099269, 0.6417152113828789] | [0.9765905703089925, 0.7402706100619615] |
+| 0.1292        | 21.0  | 10794 | 0.1565          | 0.7788   | 0.8416        | 0.9431           | [0.9372341895291594, 0.6204583221235739] | [0.9795794314033889, 0.7035825637871398] |
+| 0.1284        | 22.0  | 11308 | 0.1487          | 0.7879   | 0.8532        | 0.9450           | [0.9391595949869292, 0.6366277680932542] | [0.9780565365202, 0.7282789363894756]    |
+| 0.1279        | 23.0  | 11822 | 0.1461          | 0.7927   | 0.8629        | 0.9456           | [0.9395641382795358, 0.6458609695526066] | [0.9752807298744423, 0.7506032283294566] |
+| 0.1262        | 24.0  | 12336 | 0.1436          | 0.7934   | 0.8633        | 0.9458           | [0.9398055111519206, 0.6469847970011983] | [0.9754426177792707, 0.7512221554580607] |
+| 0.1223        | 25.0  | 12850 | 0.1465          | 0.7945   | 0.8622        | 0.9464           | [0.9404797675363489, 0.6484203969264617] | [0.9766238155760367, 0.7478641586538628] |
+| 0.1234        | 26.0  | 13364 | 0.1435          | 0.7925   | 0.8570        | 0.9463           | [0.9405453729701521, 0.6444525239879496] | [0.9784625404961454, 0.735513574144182]  |
+| 0.1223        | 27.0  | 13878 | 0.1464          | 0.7937   | 0.8618        | 0.9461           | [0.9402290147962161, 0.6472276210560935] | [0.976462283595653, 0.7471743581526247]  |
+| 0.1196        | 28.0  | 14392 | 0.1450          | 0.7929   | 0.8589        | 0.9462           | [0.9403789254222912, 0.645380249186489]  | [0.9776359685007636, 0.7400721898628863] |
+| 0.1199        | 29.0  | 14906 | 0.1451          | 0.7919   | 0.8563        | 0.9462           | [0.9403753807784087, 0.6433302809940281] | [0.9784796056303284, 0.7341607320998506] |
+| 0.1208        | 30.0  | 15420 | 0.1438          | 0.7943   | 0.8621        | 0.9463           | [0.9404316292856895, 0.6481512759943856] | [0.9766097421311045, 0.7476136089750051] |
 
 
 ### Framework versions