Solved by ptrblck in post #2 
              
              
                If I understand the workflow correctly, you are now feeding the raw logits to nn.BCEWithLogitsLoss, which yields the same loss as in the float32 training, and use the output = torch.sigmoid(model(images)) tensor only to visualize the samples? 
If so, do you see any difference by converting the outp…