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,