Update docstring

ashleve · Dec 1, 2023 · 27fd562 · 27fd562
1 parent 473cfbf
commit 27fd562
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/models/mnist_module.py b/src/models/mnist_module.py
@@ -51,6 +51,7 @@ def __init__(
         :param net: The model to train.
         :param optimizer: The optimizer to use for training.
         :param scheduler: The learning rate scheduler to use for training.
+        :param compile: Whether to compile the model before training.
         """
         super().__init__()