机器学习PAI如果我想训练评估后直接导出,应该改哪个配置呢?如何让chief导出呢?我该改哪里啊?[阿里云机器学习PAI]

机器学习PAI如果我想训练评估后直接导出,应该改哪个配置呢?如何让chief导出呢?我该改哪里啊?

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
=====这是一个广告位,招租中,联系qq 78315851====
2 条回复 A 作者 M 管理员
  1. 在阿里云的机器学习PAI(Platform for AI Innovation)中,如果你想在训练评估后立即导出模型,你可以使用--eval_and_export选项。这个选项会在每次训练迭代结束后立即执行export操作,将当前的模型参数保存到文件中。

    如果你想让chief节点导出模型,你需要确保你的任务配置中包含了chief节点。然后在export_config字段中指定导出模型的路径和名称。

    以下是一个示例的任务配置:

    {  "name": "my_model",  "task_type": "train",  "data_config": "path/to/your/data/config",  "model_config": "path/to/your/model/config",  "run_config": "path/to/your/run/config",  "export_config": {    "output_path": "/path/to/your/output/directory",    "output_filename": "my_model"  },  "resource_config": "path/to/your/resource/config",  "job_config": "path/to/your/job/config",  "eval_and_export": true}

    在这个配置中,eval_and_export选项被设置为true,表示每次训练迭代结束后都会执行export操作。export_config字段指定了导出模型的路径和名称。

  2. 代码里面 在_train_and_evaluate完了之后,调一下estimator.export即可,此回答整理自钉群“【EasyRec】推荐算法交流群”