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