更新时间:2023-11-17 GMT+08:00
训练作业的启动文件如何获取训练作业中的参数?
训练作业参数有两种来源,包括后台自动生成的参数和用户手动输入的参数。具体获取方式如下:
- 创建训练作业时,训练作业运行参数中的“train_url”指的是训练输出位置,“data_url”指的是选择的数据来源路径。“test”的参数为手动输入。
图1 创建训练作业(新版)
图2 创建训练作业(旧版)
- 训练作业运行成功之后,在训练作业列表中,您可以单击作业名称,查看该作业的详情。可从日志中获取参数的传入方式,如图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