From 89daa407e5a231d78f35bde36c2bfbef8546cdc5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Guillermo=20Garc=C3=ADa?= Date: Wed, 5 Dec 2018 14:40:34 +0100 Subject: [PATCH] train.py report argument as store_true Default is false: python train.py If want the report: python train.py --report --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 38f86035..35f7d52b 100644 --- a/train.py +++ b/train.py @@ -225,7 +225,7 @@ if __name__ == '__main__': parser.add_argument('--multi-scale', default=False, help='random image sizes per batch 320 - 608') parser.add_argument('--img-size', type=int, default=32 * 13, help='pixels') parser.add_argument('--resume', action='store_true', help='resume training flag') - parser.add_argument('--report', default=False, help='report TP, FP, FN, P and R per batch (slower)') + parser.add_argument('--report', action='store_true', help='report TP, FP, FN, P and R per batch (slower)') parser.add_argument('--freeze-darknet53', default=False, help='freeze darknet53.conv.74 layers for first epoch') parser.add_argument('--var', type=float, default=0, help='optional test variable') opt = parser.parse_args()