From eaa14272c205bae3a6b2861f330cf1709dd5f670 Mon Sep 17 00:00:00 2001 From: Shawn Cicoria Date: Fri, 13 Nov 2020 13:25:19 -0500 Subject: [PATCH] correct the --save option currently the logic for --no-save and checking the flag is inverted -- running without the flag doesn't save, and using --no-save essentially enforces same. change to --save and store as TRUE --- bit_pytorch/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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()) +