diff --git a/npu/helloworld.py b/npu/helloworld.py
index 410d849..1af4e33 100644
--- a/npu/helloworld.py
+++ b/npu/helloworld.py
@@ -6,7 +6,7 @@ import torch_npu  # 确保安装了 torch_npu
 
 # 创建参数解析器
 parser = argparse.ArgumentParser(description='Training script with output path option')
-parser.add_argument('--input', default=None, type=str, help='input, where the dataset is stored.')
+parser.add_argument('--dataset_input', default=None, type=str, help='dataset_input, where the dataset is stored.')
 parser.add_argument('--output', type=str, default='', help='Output path for saving the result')
 args = parser.parse_args()
 
@@ -46,4 +46,4 @@ try:
     mox.file.copy(local_result_path, obs_result_path)
     print(f"Result has been successfully saved to {obs_result_path}")
 except Exception as e:
-    print(f"Failed to save result to OBS: {e}")
+    print(f"Failed to save result to OBS: {e}")
\ No newline at end of file