大家好,欢迎来到IT知识分享网。
当我们需要把库当脚本运行的时候,就需要使用python -m了。举个栗子:
使用pytorch进行分布式模型运行swin transformer的时候:
python -m torch.distributed.launch --nproc_per_node 1 --master_port 12345 main.py --eval --cfg configs/swin_tiny_patch4_window7_224.yaml --resume models/swin_tiny_patch4_window7_224.pth --data-path imagenet/
咱们很容易通过命令行跑通上面的脚本,但是不方便调试。在pycharm中调试就需要能在pycharm中运行起来。
首先要知道的是:我们使用launch来出发main,py,对pycharm而言,这里运行的主要脚本是launch而不是main.py。
所以需要找到launch脚本的位置,随便找个.py文件在pycharm中打开。输入:
import torch.distributed.launch
用ctrl+单击launch就可以进入到launch脚本所在的目录位置。
比如:
/home/username/anaconda3/lib/python3.7/site-packages/torch/distributed/launch.py
在configurations中配置好Script path的位置:
然后把后面的东西都复制到Parameters那一行:
--nproc_per_node 1 --master_port 12345 main.py --eval --cfg configs/swin_tiny_patch4_window7_224.yaml --resume models/swin_tiny_patch4_window7_224.pth --data-path imagenet/
别急,还有最后一步,需要更改路径。因为你现在的默认路径已经是launch.py的那个路径,然而你的main.py并不在那个路径。
我们看到Working directory那一行,改成你main.py所在的路径就可以了~
有问题欢迎留言交流~
免责声明:本站所有文章内容,图片,视频等均是来源于用户投稿和互联网及文摘转载整编而成,不代表本站观点,不承担相关法律责任。其著作权各归其原作者或其出版社所有。如发现本站有涉嫌抄袭侵权/违法违规的内容,侵犯到您的权益,请在线联系站长,一经查实,本站将立刻删除。 本文来自网络,若有侵权,请联系删除,如若转载,请注明出处:https://yundeesoft.com/24404.html