Image segmentation code on personal laptop

I tried running the image segmentation notebook on my personal laptop, after downloading all the images etc. Everything was running well until I got to the train model section and got the below error, not sure why:

Thanks for your help

code below for completeness: