更新 'inference.py'

2022-07-13 15:20:56 +08:00 · 2022-07-13 15:20:56 +08:00 · ec7c16f505
parent df7c09155a
commit ec7c16f505
1 changed files with 1 additions and 1 deletions
--- a/inference.py
+++ b/inference.py
@ -144,7 +144,7 @@ parser.add_argument('--device_target', type=str, default="Ascend", choices=['Asc
                    help='device where the code will be implemented (default: Ascend)')                

 if __name__ == "__main__":            
-    args = parser.parse_args()
+    args, unknown = parser.parse_known_args()
    ### defining the training environment
    environment = 'train'
    workroot = WorkEnvironment(environment)