U-net training Error: The size of tensor a (16) must match the size of tensor b (6) at non-singleton dimension 1

Question

I’m trying to train a Unit model on LandCoverNet dataset, which is a satellite imagery dataset that contains input images and corresponding land cover type masks. I have cre