在pycharm编辑器里跑swin-transformer

swin-transformer在终端执行的命令的是：

python -m torch.distributed.launch --nproc_per_node 1 --master_port 12345 main.py --cfg configs/swin/swin_tiny_patch4_window7_224.yaml --data-path imagenet --batch-size 2

参数解析：

python -m torch.distributed.launch 表示调用torch.distributed.launch这个.py文件进行分布式训练；

--nproc_per_node ：表示创建节点数，这个值通常与训练使用的GPU数量一致。

 

把在终端运行转到pycharm中运行：

1. 建立distributed模块与swin-transformer文件夹的软连接：

ln -s /home/azhihong/anaconda3/envs/swin/lib/python3.7/site-packages/torch/distributed/ /home/azhihong/Swin-Transformer/

2. 打开pycharm，依次点击Run->Edit Configurations 进入参数配置界面，修改script path和parameters。

其中Script path 和 Parameters的参数为：

script path: /home/azhihong/Swin-Transformer/distributed/launch.py

Parameters: --nproc_per_node 1 --master_port 12345 /home/azhihong/Swin-Transformer/main.py --eval --cfg /home/azhihong/Swin-Transformer/configs/swin/swin_tiny_patch4_window7_224.yaml --data-path /home/azhihong/Swin-Transformer/imagenet --batch-size 2