Fix loading MNISTLitModule from ckpt (#481)

ashleve · Dec 17, 2022 · 75b44ff · 75b44ff
1 parent 28386d6
commit 75b44ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/models/mnist_module.py b/src/models/mnist_module.py
@@ -31,7 +31,7 @@ def __init__(
 
         # this line allows to access init params with 'self.hparams' attribute
         # also ensures init params will be stored in ckpt
-        self.save_hyperparameters(logger=False, ignore=["net"])
+        self.save_hyperparameters(logger=False)
 
         self.net = net