diff --git a/train.py b/train.py index 84932406..7548e4e1 100644 --- a/train.py +++ b/train.py @@ -11,7 +11,7 @@ parser.add_argument('-epochs', type=int, default=100, help='number of epochs') parser.add_argument('-batch_size', type=int, default=2, help='size of each image batch') parser.add_argument('-data_config_path', type=str, default='cfg/coco.data', help='data config file path') parser.add_argument('-cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path') -parser.add_argument('-multi_scale', default=True, help='random image sizes per batch 320 - 608') +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', default=False, help='resume training flag') parser.add_argument('-batch_report', default=False, help='report TP, FP, FN, P and R per batch (slower)')