更新时间: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") # 通过工程的根目录得到依赖文件路径
父主题: 编写训练代码