← Questions
Debug a shape mismatch in a PyTorch U-Net
0
I was asked this
No reports yet. Sign in to say where you were asked it.
A U-Net written in PyTorch fails with a tensor shape mismatch. Find the cause (typically where skip connections are concatenated with upsampled features whose size no longer matches after pooling and convolutions), fix it, and explain how you would make the model work for any input size.
No official answer yet. Write yours on the right and run it.
Nothing published yet.
Sign in to submit your code; every attempt is kept here.
Write and test your solution herePython in your browser, the question's tests level by level, every submission kept for you.Sign in to start coding