Update model.py

layumi · web-flow · commit 988db34f589f · 2025-04-21T17:31:20.000+08:00
diff --git a/model.py b/model.py
@@ -41,7 +41,7 @@ def __init__(self, kernel_size=3, padding=1, polish=True):
 
         self.polish = polish
         self.pad = padding
-        self.relu = nn.ReLU()
+        self.relu = nn.LeakyReLU(0.1)
         self.bn = nn.BatchNorm2d(1)
 
     def __call__(self, x):
