背景介绍

接着上次的vision mamba初步跑通， 想进一步了解内部代码运行的过程， 模型的工作机理，因此打算利用pycharm进行断点调试(即debug)， 花了半天时间终于学会了如何将控制台命令的形式传入pycharm中进行传参。在此， 感谢大佬的博客, 这里只是具体结合本机安装wsl2对大佬的博客进行了复现， 如果是远程服务器，可能看大佬的博客更合适。

进行配置

1. 打开pycharm工程

2. 对python解释器进行配置

红色位置这里点一下， 一般是需要你进行选择的， 然后是这里， 点击

进行下一步，进入到这里

找到这里\home\用户名\anaconda3\envs\mamba\bin\python3.10点击创建。

3. 分布式debug的配置

找到Vim/vim/main.py, 先运行一下， 就会出现类似于这里的图标， 只不过你们这里是main， 而不是launch。

接下来按照图片操作

打开配置后进行修改， 这里是修改过的， 最下面红色的圈那里需要添加CUDA的device

然后修改script这里的配置位置：
这个路径位置需要一直往下划， 在这个位置：

位置在这里(以我的为例， 具体的位置需要结合你的用户名和虚拟环境名字查找)：

最后添加之前命令行的参数：
在这里

输入以下内容：

--master_port=6666
--nproc_per_node=1
main.py
--model
vim_small_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2
--batch-size
32
--drop-path
0.05
--weight-decay
0.05
--lr
1e-3
--num_workers
1
--data-set
CIFAR
--data-path
./data
--output_dir
./output/vim_small_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2
--no_amp

然后就可以打断点进行debug调试了！！！

参考资料

1. https://blog.csdn.net/liu_yuan_kai/article/details/118706387?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522171541553016800180641921%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=171541553016800180641921&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2_allfirst_rank_ecpm_v1~rank_v31_ecpm-4-118706387-null-null.142^v100pc_search_result_base2&utm_term=pycharm%20debug%20torchrun&spm=1018.2226.3001.4187