From d03ce45da5e83527c6225fecd0b17d96887fdf6c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Guillermo=20Garc=C3=ADa?= Date: Wed, 5 Dec 2018 15:55:09 +0100 Subject: [PATCH] train.py freeze-darknet53 shortened to freeze and action store_true Traing with freeze: python train.py --freeze Train without freeze: python train.py Note: in the actual version freeze is only for first epoche --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 5b0ff69a..cfe90dea 100644 --- a/train.py +++ b/train.py @@ -247,7 +247,7 @@ if __name__ == '__main__': parser.add_argument('--weights-path', type=str, default='weights', help='path to store weights') parser.add_argument('--resume', action='store_true', help='resume training flag') 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('--freeze', action='store_true', help='freeze darknet53.conv.74 layers for first epoche') parser.add_argument('--var', type=float, default=0, help='optional test variable') opt = parser.parse_args() print(opt, end='\n\n') @@ -265,6 +265,6 @@ if __name__ == '__main__': weights_path=opt.weights_path, report=opt.report, multi_scale=opt.multi_scale, - freeze_backbone=opt.freeze_darknet53, + freeze_backbone=opt.freeze, var=opt.var, )