main_train_process
接口功能
trainer函数入口。除了用于存储中间状态的工作目录参数是s3路径外,其他的输入都是从本地文件系统输入。
接口引用路径
ockics.entry.main_train_process
接口格式
函数:main_train_process(train_para)
输入参数
参数 |
类型 |
是否必选 |
说明 |
取值要求 |
---|---|---|---|---|
train_para |
ParaTrain |
必选 |
trainer组件的所有输入数据。 |
函数参数,元组类型,具体元素详见ParaTrain。 |
返回值
无
使用样例
import argparse from collections import namedtuple ParaTrain = namedtuple("ParaTrain", ["base_model", "train_datasets", "train_dataset_images", "eval_datasets", "eval_dataset_images", "train_config", "dataset_fmt", "standalone", "work_url", "output_train_artifact", "output_infer_artifact", "s3_endpoint", "s3_ak", "s3_sk", "s3_secure", "s3_certcheck"]) def main_train(s3_endpoint=None, s3_ak=None, s3_sk=None, s3_secure=None, s3_certcheck=None): argss = _parse_args() para = ParaTrain(argss.baseModel, argss.trainDatasets, argss.train_dataset_images, argss.evalDatasets, argss.eval_dataset_images, argss.trainConfig, argss.dataset_fmt, argss.standalone, argss.workurl, argss.outputTrainArtifact, argss.outputInferArtifact, s3_endpoint, s3_ak, s3_sk, s3_secure, s3_certcheck) main_train_process(para)
父主题: trainer组件接口