|
|
@@ -467,7 +467,7 @@ if __name__ == '__main__':
|
|
|
parser.add_argument('--cfg', type=str, default='models/yolov5s.yaml', help='model.yaml path')
|
|
|
parser.add_argument('--data', type=str, default='data/voc.yaml', help='data.yaml path')
|
|
|
parser.add_argument('--hyp', type=str, default='data/hyp.scratch.yaml', help='hyperparameters path')
|
|
|
- parser.add_argument('--epochs', type=int, default=5)
|
|
|
+ parser.add_argument('--epochs', type=int, default=50)
|
|
|
parser.add_argument('--batch-size', type=int, default=8, help='total batch size for all GPUs')
|
|
|
parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='[train, test] image sizes')
|
|
|
parser.add_argument('--rect', action='store_true', help='rectangular training')
|
