AI开发平台MODELARTS-训练作业的启动文件如何获取训练作业中的参数？

时间：2024-04-30 17:35:42

AI开发平台MODELARTS

训练作业的启动文件如何获取训练作业中的参数？

训练作业参数有两种来源，包括后台自动生成的参数和用户手动输入的参数。具体获取方式如下：

创建训练作业时，“输入”支持配置训练的输入参数名称（一般设置为“data_url”），以及输入数据的存储位置，“输出”支持配置训练的输出参数名称（一般设置为“train_url”），以及输出数据的存储位置。
训练作业运行成功之后，在训练作业列表中，您可以单击作业名称，查看该作业的详情。可从日志中获取参数的传入方式，如图1所示。
图1 查看日志

如果需在训练中获取“train_url”、“data_url”和“test”参数的值，可在训练作业的启动文件中添加以下代码获取：

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--data_url', type=str, default=None, help='test')
parser.add_argument('--train_url', type=str, default=None, help='test')
parser.add_argument('--test', type=str, default=None, help='test')

父主题： 编写训练代码

上一篇：AI开发平台MODELARTS-如何在代码中打印GPU使用信息:使用python命令

下一篇：AI开发平台MODELARTS-远程连接处于retry状态如何解决？:解决方法