dist_train.sh转为launch.json启动项目

最新推荐文章于 2024-07-04 14:36:56 发布

小菜鸡本航

最新推荐文章于 2024-07-04 14:36:56 发布

阅读量440

点赞数 1

CC 4.0 BY-SA版权

文章标签： json

本文链接：https://blog.csdn.net/hu_yuhang/article/details/132182342

本文介绍了如何在VSCode中使用launch.json文件替代bash脚本来启动Python分布式训练，特别关注了如何将原始bash环境变量转换为JSON格式的配置，以解决PYTHONPATH问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用vscode进行debug时，需要写一个launch.json文件来启动，这就需要将原始的bash文件中的变量配置转换为json配置。

原始bash指令如下

#!/usr/bin/env bash
CONFIG=$1

PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
python -m torch.distributed.launch \
    $(dirname "$0")/train.py \
    $CONFIG \
    --seed 0 \
    --launcher pytorch ${@:3}

launch.json如下：program里面使用launch.py 对应 python -m torch.distributed.launch。env一定要填写pythonpath，否则会出现找不到模块的情况。

{
    // Use IntelliSense to learn about possible attributes.
    // Hover to view descriptions of existing attributes.
    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
    "version": "0.2.0",
    "configurations": [
        {
            "name": "Python: Current File",
            "type": "python",
            "request": "launch",
            "program": "/root/miniconda3/envs/occformer/lib/python3.7/site-packages/torch/distributed/launch.py",
            "console": "integratedTerminal",
            "args": [
                "tools/train.py",
                "--config",
                "/root/OccFormer/projects/configs/occformer_nusc/occformer_nusc_panoptic_r50_256x704.py"
            ],
            "env": {"PYTHONPATH": "\"$(dirname $0)/..\":${env:PYTHONPATH}"},
        }
    ]
}