From 5a5eeb98e2228b35eca70d8b6ce7162c0d687efd Mon Sep 17 00:00:00 2001 From: "zzysz@qq.com" Date: Fri, 5 Sep 2025 09:09:16 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E7=9A=84=E6=B5=8B=E8=AF=95=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E5=92=8C=E8=80=81=E7=9A=84=E6=B5=8B=E8=AF=95=E4=BB=A3?= =?UTF-8?q?=E7=A0=81=E6=9B=B4=E6=96=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- npu/helloworld.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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