From 0eb25ed62de97c43a24c50b3d7100304254102bb Mon Sep 17 00:00:00 2001 From: andreped Date: Sun, 21 Aug 2022 20:50:19 +0200 Subject: [PATCH] change default batch size to 512 --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 8796fcd..de80639 100644 --- a/main.py +++ b/main.py @@ -21,7 +21,7 @@ help="which model to use for deployment (give full path, without extension '.zip'.") parser.add_argument('--infer_pversion', metavar='-py', type=str, nargs='?', default="3.7", help="define which python version the current deployment model is trained with.") - parser.add_argument('--batch_size', metavar='--bs', type=int, nargs='?', default=32, + parser.add_argument('--batch_size', metavar='--bs', type=int, nargs='?', default=512, help="set which batch size to use for training.") parser.add_argument('--learning_rate', metavar='--lr', type=float, nargs='?', default=0.0003, help="set which learning rate to use for training.")