diff --git a/train.py b/train.py index a4af4af6..493a28fd 100644 --- a/train.py +++ b/train.py @@ -7,7 +7,7 @@ from utils.datasets import * from utils.utils import * parser = argparse.ArgumentParser() -parser.add_argument('-epochs', type=int, default=999, help='number of epochs') +parser.add_argument('-epochs', type=int, default=160, help='number of epochs') parser.add_argument('-batch_size', type=int, default=12, 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')