深度学习框架日常踩坑记录
pytorch,tensorflow,caffe的日常踩坑记录
FXY.jpg
这个作者很懒,什么都没留下…
展开
-
Numpy介绍
一、Numpy是什么?Numpy是python的扩展程序库,支持大量的维度与矩阵的运算,就是进行数组运算的库。Numpy一个非常重要的特点就是用于存放同类型元素N维数组的ndarray对象,由下面几个组成。1、指向数据的指针2、数据类型,描述数组中固定大小值的格子3、表示数组形状的元组4、跨度元组,其中的整数指的是为了前进到当前维度下一个元素需要"跨过"的字节数。注:有了列表为什么还...原创 2019-03-13 19:30:13 · 257 阅读 · 0 评论 -
pytorch训练自己的数据集
将图片放在每类的文件夹下面,接着放一个txt文件使用img_data = torchvision.datasets.ImageFolder(image_path, transform)原创 2019-05-25 08:57:27 · 3802 阅读 · 0 评论 -
pytorch在无anaconda环境下源码编译安装
参考自源码编译pytorch由于要紧急测试pytorch的代码,实验室环境下无anaconda,打算直接在无anaconda的环境下编译pytorch的源码1、首先安装yaml依赖sudo apt install python-yaml python3-yaml2、git clone pytorch 的源码git clone https://github.com/pytorch/pyt...原创 2019-03-13 19:55:29 · 952 阅读 · 1 评论 -
关于pytorch中optimizer.zero_grad()
这个查了好多资料才搞明白一个Batch有最少一张图片,计算loss的时候是对Batch_size张图片的loss对weight的导数的平均数,所以会有一个Batch_size张图片loss累加的计算的过程,这时候在计算新的导数的时候,是要进行一次清零才能计算新一轮Batch中Batch_size张图片的导数...原创 2019-03-27 17:19:40 · 2760 阅读 · 0 评论 -
解决KeyError: 'unexpected key "module.encoder.embedding.weight" in state_dict'
原因:可能是之前训练的时候开了nn.DataParallel,造成了储存模型的时候是放在module里面的。解决方案:1、关掉nn.DataParallel 重新训练权重文件2、模型参数文件是以OrderDict储存,可以手动去除尝试一下3、新的模型训练的时候开启nn.DataParallel尝试一下...原创 2019-03-20 09:19:46 · 2118 阅读 · 0 评论 -
tensorflow,keras安装
pip install tensorflow-gpu==1.2注:cuda是8.0,cudnn版本也比较低,所以安装tensorflow的版本比较低,不然回报错ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory...原创 2018-10-24 20:42:39 · 85 阅读 · 0 评论 -
解决CAFFE编译过程中的问题
首先确保依赖全部装好`sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compilersudo apt-get install --no-install-recommends libboost-all-devsudo apt...原创 2018-10-13 22:15:03 · 4039 阅读 · 0 评论