Browse Source

新的测试数据和老的测试代码更新

master
zzysz@qq.com 2 months ago
parent
commit
5a5eeb98e2
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      npu/helloworld.py

+ 2
- 2
npu/helloworld.py View File

@@ -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}")

Loading…
Cancel
Save