分类: Python

9 篇文章

内存一直增加—-代码里有matplotlib绘图的代码
问题描述 最近在调试代码的时候发现内存一直在增加,持续一段时间后程序就会停止,但是排查了好久也没有发现问题. 最后发现竟然是绘图代码的问题,没有plt.close(),导致生成的绘图一直保存在内存里不断增加. 解决方案 增加一行代码plt.close(),把绘图关闭
thumbnail
模型训练—-parser.add_argument添加配置参数
现在需要配置参数来达到修改训练的方式,我现在需要新建一个参数来开关wandb的使用。 首先就是在def parse_option():函数里添加上你要使用的变量名 parser.add_argument("--open_wandb",type= bool,default=False,help='open wandb&#…
Pycharm—-将Anaconda建立的环境导入
首先打开项目设置,点击添加 随后点击现有环境,点击三个。。。号进行添加 最后找到你Anaconda安装文件夹,envs找到你建立的环境名称,找到python.exe将它导入即可让现在的python环境为你建立的环境,同时还需要更改终端方便pip安装库 更改终端看这篇
Python—-多线程使用
在运行代码的时候,总会想让程序同时运行几个东西,这个时候就需要用到多线程。可以说使用python中的threading库来实现多线程,多线程运行的具体线程数量和每个线程的执行时间由操作系统的调度策略决定。 比如我现在需要同时发起请求,那么就可以发起请求的业务带啊吗分装为一个函数,放入线程池thread = threading.Thread(targ…
Vscode—-远程服务器改名
问题描述 一开始Autodl服务器机子很多,但是我使用vscode的时候他们的名字都一样,导致每次要打开机子是都需要重新输入ssh和密码 解决方法 修改vscode端服务器的名字即可解决 打开远程设置,选择你的ssh配置文件 将Host改为你想要的名字,保存刷新即可
模型训练—-apex库报错IndexError: tuple index out of range
问题描述 在训练模型的过程中遇到了apex库的报错IndexError: tuple index out of range导致无法训练。在github查询后找到了解决方法 问题解决 需要修改/apex-master/apex/amp/utils.py这个文件的代码 从93行开始修改 if x in cache: cached_x = cache[x…
thumbnail
模型训练—-将pth模型转换为onnx
Github代码 1 安装需要的环境 需要在虚拟环境中安装onnx和onnxruntime(GPU),环境和自己的cuda版本要对应上查询链接 激活环境,查看环境的cuda版本,我是cuda11.6 +cudnn8302,那就选择1.14吧 conda activate xxxx python import torch # 查询cuda版本 pri…
thumbnail
模型训练—-加载自己的数据集,train.txt包含“路径,类别”
今天想训练一个restnet,之前我是会用文件夹的方式加载数据集,但是师兄给我的数据是用.txt文件划分的训练集和验证集,知道是要把路径和标签提取出来,搞了半天没搞成,后边找到咋搞了。 我的数据形式是每个类别一个文件夹,路径和类别存到.txt里 首先解决中文类别的问题,将中文类别单独新建一个txt,按顺序给他们赋予数字,在对应起来 # 读取类别文件…
thumbnail
VScode—-debug调试python代码添加上额外命令(args)
问题描述 在服务器上运行python代码时,总会添加上额外的参数一般是用jyputer或者终端直接加上命令,现在我在vscode调试远程代码的时候想要加上这些命令. 问题解决 打开要调试的主函数打上断点,启动调试 此时会报错,我们点击添加配置,修改自动生成的lauch.json 将对应的参数存入args即可,"","&…

超多性价比流量卡,扫码查看

这将关闭于 20