Fix the shape assertion error in the evaluation stage of one-class se…#446

littsk wants to merge 1 commit intomilesial:mastermilesial/Pytorch-UNet:masterfrom

littsk:bug_fixlittsk/Pytorch-UNet:bug_fixCopy head branch name to clipboard