- 博客(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
3025
5
原创 win10+python3.5 图像标注工具LabelImg的安装
1. LabelImg下载地址 https://github.com/tzutalin/labelImg download下来，解压 2. 安装必备的python包 PyQt5 PyQt5_tools lxml 安装方法： ①pip install xxx ②到https://www.lfd.uci.edu/~gohlke/pythonlibs/#gdal下下载相应的.whl 3.pyr...
2020-02-20 13:22:24
215
原创 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
1741
原创 argparse的简单用法总结
argparse是python自带的命令行参数解析包 ，自己也简单总结了一下 一、基本框架 import argparse # 导入命令行解析库 parser = argparse.ArgumentParser() # 声明一个命令行解析器 parser.add_argument() # 添加参数 args = parser.parse_args() # 利用parse_args获取所有参数 ...
2020-02-16 10:48:46
372
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人