diff --git a/inference.py b/inference.py
index f99d055..f0501e9 100644
--- 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)