diff --git a/trainer.py b/trainer.py
index 92b15a9a..e0941402 100644
--- a/trainer.py
+++ b/trainer.py
@@ -48,7 +48,7 @@ def cli_main(
                 mode="min",
                 save_last=True,
                 save_top_k=2,
-                save_weights_only=True,
+                save_weights_only=False,
                 verbose=True,
             ),
             LearningRateMonitor(logging_interval="step"),