[MMEditing] MMEditing多线程vscode调试

NK_ZZL

于 2021-05-06 10:14:15 发布

阅读量527

点赞数

分类专栏： Pytorch 文章标签： python debug

本文链接：https://blog.youkuaiyun.com/qq_42263831/article/details/116448042

版权

Pytorch 专栏收录该内容

5 篇文章

订阅专栏


{
    // 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: MMEditing",
            "type": "python",
            "request": "launch",
            "program": "/home/vm411/anaconda3/lib/python3.8/site-packages/torch/distributed/launch.py",
            "console": "integratedTerminal",
            "args": [
                "--nproc_per_node=1",
                "--master_port=49500",
                "tools/train.py",
                "./configs/restorers/basicvsr/basicvsr_reds4_nb15.py",
            ],
            "env": {"CUDA_VISIBLE_DEVICES":"0"},
        }
    ]
}

MMEditing采用分布式训练，调试过程与一般pytorch程序有所区别，需要指定分布式launch，将代码路径作为参数输入。