From 05f28ab02b591e5c95d1ddae42d6de74e302be3b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 15 Oct 2018 21:05:24 +0200 Subject: [PATCH] -batch_size from 12 to 16 --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 5ea4741d..5a746407 100644 --- a/train.py +++ b/train.py @@ -7,7 +7,7 @@ from utils.utils import * parser = argparse.ArgumentParser() parser.add_argument('-epochs', type=int, default=68, help='number of epochs') -parser.add_argument('-batch_size', type=int, default=12, help='size of each image batch') +parser.add_argument('-batch_size', type=int, default=16, 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('-img_size', type=int, default=32 * 13, help='size of each image dimension')