diff --git a/train.py b/train.py index 6336ec7..f7128ae 100644 --- a/train.py +++ b/train.py @@ -15,7 +15,7 @@ def parse_arguments(): p.add_argument('-epochs', type=int, default=100, help='number of epochs for train') p.add_argument('-batch_size', type=int, default=32, - help='number of epochs for train') + help='batch size for training') p.add_argument('-lr', type=float, default=0.0001, help='initial learning rate') p.add_argument('-grad_clip', type=float, default=10.0,