Removed default parameter

SINGROUP · Feb 7, 2024 · 59a225e · 59a225e
1 parent b03f06b
commit 59a225e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlspm/image/models.py b/mlspm/image/models.py
@@ -52,7 +52,7 @@ class AttentionUNet(nn.Module):
 
     def __init__(
         self,
-        z_in: int = 10,
+        z_in: int,
         n_in: int = 1,
         n_out: int = 3,
         in_channels: int = 1,