AI开发平台MODELARTS-开发自定义脚本:新旧版训练代码适配的差异

时间：2024-01-04 11:05:16

AI开发平台MODELARTS

新旧版训练代码适配的差异

旧版训练中，用户需要在输入输出数据上做如下配置：

#解析命令行参数
import argparse
parser = argparse.ArgumentParser(description='MindSpore Lenet Example')
parser.add_argument('--data_url', type=str, default="./Data",
                    help='path where the dataset is saved')
parser.add_argument('--train_url', type=str, default="./Model", help='if is test, must provide\
                    path where the trained ckpt file')
args = parser.parse_args()
...
#下载数据参数至容器本地，在代码中使用local_data_path代表训练输入位置
mox.file.copy_parallel(args.data_url, local_data_path)
...
#上传容器本地数据至obs路径
mox.file.copy_parallel(local_output_path, args.train_url)

上一篇：DDOS防护 AAD-配置DDoS高防日志:日志字段说明

下一篇：AI开发平台MODELARTS-查看训练作业资源利用率:如何提高训练作业资源利用率