文档首页> AI开发平台ModelArts> 常见问题> 训练作业> 编写训练代码> 训练作业的启动文件如何获取训练作业中的参数?
更新时间:2024-04-30 GMT+08:00
分享

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

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

  1. 创建训练作业时,“输入”支持配置训练的输入参数名称(一般设置为“data_url”),以及输入数据的存储位置,“输出”支持配置训练的输出参数名称(一般设置为“train_url”),以及输出数据的存储位置。
  2. 训练作业运行成功之后,在训练作业列表中,您可以单击作业名称,查看该作业的详情。可从日志中获取参数的传入方式,如图1所示。
    图1 查看日志
  3. 如果需在训练中获取“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')
分享:

编写训练代码 所有常见问题

more