添加参数类型

test.py

@@ -108,8 +108,8 @@ if __name__ == "__main__":
     parser.add_argument('--batch_size', default=100, type=int, help='testing batch size.')
     parser.add_argument('--cuda', default=True, type=bool, help='Use cuda to train model')
     parser.add_argument('--device', default='cuda', help='Use cuda to vaildate model')
-    parser.add_argument('--export_onnx', default=False, help='export model to onnx')
-    parser.add_argument('--dynamic', default=False, help='use dynamic batch size')
+    parser.add_argument('--export_onnx', default=False, type=bool, help='export model to onnx')
+    parser.add_argument('--dynamic', default=False, type=bool, help='use dynamic batch size')
     parser.add_argument('--phase_train', default=False, type=bool, help='train or test phase flag.')
     parser.add_argument('--num_workers', default=8, type=int, help='Number of workers used in dataloading')