From 70ba7f0805da0ef9dc72721f1b69d4a8b0a957fb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 30 May 2019 19:08:43 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index c4f0181b..5aaa48b1 100644 --- a/train.py +++ b/train.py @@ -303,7 +303,7 @@ def print_mutation(hyp, results): if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('--epochs', type=int, default=273, help='number of epochs') + parser.add_argument('--epochs', type=int, default=68, help='number of epochs') parser.add_argument('--batch-size', type=int, default=16, help='size of each image batch') parser.add_argument('--accumulate', type=int, default=4, help='accumulate gradient x batches before optimizing') parser.add_argument('--cfg', type=str, default='cfg/yolov3-spp.cfg', help='cfg file path')