AI开发平台MODELARTS-使用MoXing时，如何进行增量训练？:“mox.run”添加增量训练参数

时间：2024-10-24 17:22:16

AI开发平台MODELARTS 更多功能咨询

“mox.run”添加增量训练参数

在完成标注数据或数据集的修改后，您可以在“mox.run”中，修改“log_dir”参数，并新增“checkpoint_path”参数。其中“log_dir”参数建议设置为一个新的目录，“checkpoint_path”参数设置为上一次训练结果输出路径，如果是OBS目录，路径填写时建议使用“obs://”开头。

如果标注数据中的标签发生了变化，在运行“mox.run”前先执行如果标签发生变化的操作。

  mox.run(input_fn=input_fn,
          model_fn=model_fn,
          optimizer_fn=optimizer_fn,
          run_mode=flags.run_mode,
          inter_mode=mox.ModeKeys.EVAL if use_eval_data else None,
          log_dir=log_dir,
          batch_size=batch_size_per_device,
          auto_batch=False,
          max_number_of_steps=max_number_of_steps,
          log_every_n_steps=flags.log_every_n_steps,
          save_summary_steps=save_summary_steps,
          save_model_secs=save_model_secs,
          checkpoint_path=flags.checkpoint_url,
          export_model=mox.ExportKeys.TF_SERVING)

上一篇：AI开发平台MODELARTS-使用MoXing时，如何进行增量训练？:如果标签发生变化

下一篇：AI开发平台MODELARTS-使用MoXing时，如何进行增量训练？:如果标签发生变化