I am trying to reproduce semantic segmentation in this blog post. I have done some minor adjustments but still can\'t get the model to train. I always get the runtimeerror: