Debug a Buggy UNet in PyTorch

This is an Apple Machine Learning Engineer phone screen round that has tripped up candidates who did not prepare.