python
NarNi
这个作者很懒,什么都没留下…
展开
-
2021-05-25
python 学习中的一些小知识点例子1:bmm( )函数使用torch.bmm(a,b)参考https://blog.csdn.net/qq_43391414/article/details/110442229a=torch.Tensor([[[3,4],[1,2]],[[3,4],[1,2]],[[3,4],[1,2]]])b=torch.Tensor([[[1,2],[3,4]],[[1,2],[3,4]],[[3,4],[1,2]]])print(a.shape)print(b.sh原创 2021-07-14 13:52:59 · 120 阅读 · 0 评论 -
模型训练速度过慢,GPU利用率低
遇到这种情况 ——深度学习首先检查自己的数据下载部分,是不是出现重复下载。这里的重复下载是指在训练源数据的部分,单独一个函数train(train_loader,target_loader)中存在数据ids与data下载,不要将其放置在真正的循环训练中(因为本人为了容易理解,将其放入循环中了导致程序运行很慢);其次检查DataLoader中的num_workers=?,我也不清楚为什么这里设置为8,会使程序运行加快。总之,将其内部参数设置为pin_memory=True, num_workers=8。原创 2021-07-14 13:49:39 · 2300 阅读 · 0 评论 -
RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have
RuntimeError: Given groups=1, weight of size [1, 1, 3, 3], expected input[1, 3, 1402, 1200] to have 1 channels, but got 3 channels instead通过翻译得知这是 因为数据读入的通道不对,应该是1通道的,但是我们所得确是3通道的。通过查看读取的数据img的信息,发现mode是RGB。但是我们需要的是一通道的灰度图,所以需要将mode模式改变一下。原代码为:o_tensor原创 2021-06-18 10:58:43 · 8332 阅读 · 3 评论 -
torch.mean()函数
torch.mean(input) 输出input 各个元素的的均值不指定任何参数就是所有元素的算术平均值指定参数可以计算每一行或者 每一列的算术平均数对于1 如:import torcha=torch.randn(3) #随机生成一个一维的矩阵b=torch.randn(1,3) #随机生成一个二维的矩阵print(a)print(b)torch.mean(a)结果tensor([ 0.1613, 0.7873, -0.3453])tensor([[ 0.5759,原创 2021-03-12 14:31:56 · 16780 阅读 · 6 评论 -
已解决 问题ModuleNotFoundError: No module named ‘tensorflow.examples.tutorials‘
已解决 问题ModuleNotFoundError: No module named ‘tensorflow.examples.tutorials’我是一名新手 想学习一下CNN 在网上找了一些python的程序 几乎入门CNN的程序都是相关MNIST数据集的。在运行时遇到了 ModuleNotFoundError: No module named ‘tensorflow.examples.tutorials’ 问题解决import tensorflow as tfprint(tf.__versi原创 2021-03-05 10:31:26 · 2235 阅读 · 1 评论