Skip to content

Commit

Permalink
change default batch size to 512
Browse files Browse the repository at this point in the history
  • Loading branch information
andreped committed Aug 21, 2022
1 parent ade86e4 commit 0eb25ed
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.")
Expand Down

0 comments on commit 0eb25ed

Please sign in to comment.