AI开发平台MODELARTS-训练代码中，如何获取依赖文件所在的路径？

时间：2024-11-06 21:52:47

AI开发平台MODELARTS

训练代码中，如何获取依赖文件所在的路径？

由于用户本地开发的代码需要上传至ModelArts后台，训练代码中涉及到依赖文件的路径时，用户设置有误的场景较多。因此推荐通用的解决方案：使用os接口得到依赖文件的绝对路径，避免报错。

以下示例展示如何通过os接口获得其他文件夹下的依赖文件路径。

文件目录结构：

project_root                   #代码根目录
 └─bootfile.py               #启动文件
 └─otherfileDirectory        #其他依赖文件所在的目录
       └─otherfile.py        #其他依赖文件

在启动文件代码中，建议用户参考以下方式获取其他依赖文件所在路径，即示例中的“otherfile_path”。

import os
current_path = os.path.dirname(os.path.realpath(__file__)) # 获得启动文件bootfile.py的路径
project_root = os.path.dirname(current_path) # 通过启动文件路径获得工程的根目录，对应ModelArts训练控制台上设置的代码目录
otherfile_path = os.path.join(project_root, "otherfileDirectory", "otherfile.py")  # 通过工程的根目录得到依赖文件路径

父主题： 编写训练代码

上一篇：AI开发平台MODELARTS-在ModelArts创建分布式训练时如何设置NCCL环境变量？

下一篇：AI开发平台MODELARTS-在ModelArts上训练模型，输入输出数据如何配置？:解析输入路径参数、输出路径参数