update srgan

models/SRGAN/srgan.py

@@ -51,7 +51,7 @@ class GeneratorResnet(nn.Module):
         out = torch.add(out1, out2)
         out = self.upsampling(out)
         out = self.conv3(out)
-        return out
+        return out.clamp(0, 1)
     
     def inference(self, x):
         """