Train neural network for semantic segmentation (deep lab V3) with pytorch in 50 lines of code

Train net semantic segmentation net using Trans10K dataset: https://xieenze.github.io/projects/TransLAB/TransLAB.html in less then 50 lines of code (note including spaces)

Full toturial that goes with the code can be find here: https://medium.com/@sagieppel/train-neural-net-for-semantic-segmentation-with-pytorch-in-50-lines-of-code-830c71a6544f