AI开发平台MODELARTS-输出:在ModelArts上运行训练作业时，如何将训练结果存储至OBS桶？

时间：2024-12-20 10:08:04

AI开发平台MODELARTS

在ModelArts上运行训练作业时，如何将训练结果存储至OBS桶？

首先用户需要在训练代码中解析输出路径参数，可通过如下代码实现。代码后续可使用train_url指代训练容器运行过程中的本地输出路径。

更多代码适配问题参考开发自定义脚本指导。更多Moxing的操作请参考Moxing手册。

import argparse  import moxing as mox   # 创建解析  
parser = argparse.ArgumentParser(description="train mnist",                                   
                                                         formatter_class=argparse.ArgumentDefaultsHelpFormatter)  # 添加参数  
parser.add_argument('--train_url', type=str, default='obs://obs-test/ckpt/mnist',                      
                                    help='the path model saved')  # 解析参数  
args, unkown = parser.parse_known_args()