文档首页/ AI开发平台ModelArts/ 常见问题/ Standard训练作业/ 编写训练代码/ 训练代码中,如何获取依赖文件所在的路径?
更新时间:2024-10-28 GMT+08:00

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

由于用户本地开发的代码需要上传至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")  # 通过工程的根目录得到依赖文件路径