ds

Paupiera · Jan 9, 2024 · f74395f · f74395f
1 parent 34d4e3a
commit f74395f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cnn_mnist/models/model.py b/cnn_mnist/models/model.py
@@ -6,7 +6,9 @@ class MyAwesomeModel(nn.Module):
 
     """My awesome model."""
 
-    def __init__(self, number_of_kernels, kernel_sizes, input_channels, xsize_input):
+    def __init__(
+        self, number_of_kernels, kernel_sizes, input_channels, xsize_input, empty=False
+    ):
         super().__init__()
 
         self.convlayers = nn.ModuleList()