Image Segmentation with U-Net

can anyone please help me how to resolve this error

Thanks in advance

Hey could you privately message me your code for this function? This so far looks correct

For anyone reading this later. Make sure you’ve filled all Nones in the code, and make sure you’ve taken a note of number and types of outputs of the earlier implemented block functions.