diff --git a/bit_pytorch/train.py b/bit_pytorch/train.py
index 4666905..3fb51f0 100644
--- a/bit_pytorch/train.py
+++ b/bit_pytorch/train.py
@@ -283,5 +283,6 @@ def main(args):
                       help="Path to the ImageNet data folder, preprocessed for torchvision.")
   parser.add_argument("--workers", type=int, default=8,
                       help="Number of background threads used to load data.")
-  parser.add_argument("--no-save", dest="save", action="store_false")
+  parser.add_argument("--save", dest="save", action="store_true")
   main(parser.parse_args())
+