diff --git a/train.py b/train.py index 1dff7feb..60fa867d 100644 --- a/train.py +++ b/train.py @@ -15,7 +15,7 @@ parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each i parser.add_argument('-resume', default=False, help='resume training flag') parser.add_argument('-batch_report', default=False, help='report TP, FP, FN, P and R per batch (slower)') parser.add_argument('-optimizer', default='SGD', help='optimizer') -parser.add_argument('-freeze_darknet53', default=True, help='freeze darknet53.conv.74 layers for first epoch') +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() print(opt)