- 博客(4)
- 收藏
- 关注
RSS订阅原创 SSD的pytorch实现训练中遇到的问题总结
运行train.py，出现了一些错误和警告。简单总结一下：一、pytorch版本问题1、UserWarning: volatile was removed and now has no effect. Use with torch.no_grad(): instead.molded_images = Variable(molded_images, volatile=True)解决方法：im...
2020-03-04 21:01:27
1674
5
原创 win10+python3.5 图像标注工具LabelImg的安装
1. LabelImg下载地址https://github.com/tzutalin/labelImg download下来，解压2. 安装必备的python包PyQt5PyQt5_toolslxml安装方法：①pip install xxx②到https://www.lfd.uci.edu/~gohlke/pythonlibs/#gdal下下载相应的.whl3.pyr...
2020-02-20 13:22:24
138
原创 pytorch模型提示超出内存RuntimeError: CUDA out of memory.
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够简单总结一下解决方法：将batch_size改小。取torch变量标量值时使用item()属性。当1方法没用时，可以在测试阶段添加如下代码： with torch.no_grad(): # 停止自动反向计算梯度https://wisdomai.xy...
2020-02-19 18:08:31
1600
原创 argparse的简单用法总结
argparse是python自带的命令行参数解析包 ，自己也简单总结了一下一、基本框架import argparse # 导入命令行解析库parser = argparse.ArgumentParser() # 声明一个命令行解析器parser.add_argument() # 添加参数args = parser.parse_args() # 利用parse_args获取所有参数...
2020-02-16 10:48:46
289
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人