修改train.py

train.py

@@ -221,15 +221,15 @@ def train(args):
 
 def get_parser():
     parser = argparse.ArgumentParser(description='parameters to train net')
-    parser.add_argument('--max_epoch', default=15, help='epoch to train the network')
+    parser.add_argument('--max_epoch', default=15, type=int, help='epoch to train the network')
     parser.add_argument('--img_size', default=[94, 24], help='the image size')
     parser.add_argument('--train_img_dirs', default="data/train", help='the train images path')
     parser.add_argument('--test_img_dirs', default="imgs", help='the test images path')
-    parser.add_argument('--dropout_rate', default=0.5, help='dropout rate.')
-    parser.add_argument('--learning_rate', default=0.1, help='base value of learning rate.')
-    parser.add_argument('--lpr_max_len', default=8, help='license plate number max length.')
-    parser.add_argument('--train_batch_size', default=64, help='training batch size.')
-    parser.add_argument('--test_batch_size', default=10, help='testing batch size.')
+    parser.add_argument('--dropout_rate', default=0.5, type=float, help='dropout rate.')
+    parser.add_argument('--learning_rate', default=0.1, type=float, help='base value of learning rate.')
+    parser.add_argument('--lpr_max_len', default=8, type=int, help='license plate number max length.')
+    parser.add_argument('--train_batch_size', default=64, type=int, help='training batch size.')
+    parser.add_argument('--test_batch_size', default=10, type=int, help='testing batch size.')
     parser.add_argument('--phase_train', default=True, type=bool, help='train or test phase flag.')
     parser.add_argument('--num_workers', default=8, type=int, help='Number of workers used in dataloading')
     parser.add_argument('--cuda', default=True, type=bool, help='Use cuda to train model')