ibrahimth

If you make the following modifications to the function `build_res_unet`, it should work properly. ``` def build_res_unet(n_input=1, n_output=2, size=256): device = torch.device("cuda" if torch.cuda.is_available() else "cpu") body = create_body(resnet18(), pretrained=True,...