Closed pskun closed 1 year ago
合并单个任务train和inference的逻辑到同一模块;
新增pipelines模块,为每一个引擎+任务单独提供一个文件;
文件名命名规则为:engine_type + "_" + task_type + ".py"
加入注册器,新增任务只要实现指定函数即可使用;
def train_pipeline(args)
def prepare_inference(engine_type, task_type, save_path)
def inference(engine_type, task_type, samples, inference_suite)
测试详情
调用脚本训练
调用脚本预测
调用api训练和预测
(成功创建任务并训练)
(成功预测)
合并单个任务train和inference的逻辑到同一模块;
新增pipelines模块,为每一个引擎+任务单独提供一个文件;
文件名命名规则为:engine_type + "_" + task_type + ".py"
加入注册器,新增任务只要实现指定函数即可使用;
def train_pipeline(args)
def prepare_inference(engine_type, task_type, save_path)
def inference(engine_type, task_type, samples, inference_suite)
测试详情
调用脚本训练
调用脚本预测
调用api训练和预测
(成功创建任务并训练)
(成功预测)