![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
Restart丶
这个作者很懒,什么都没留下…
展开
-
tensorboard: error: unrecognized arguments:
文件路径正确,为什么会出现这个问题?原因:一般出现这个问题是因为文件命名中有出现空格,tensorboard无法识别带空格文件名文件------>将文件夹中空格部分换成下划线即可。原创 2021-03-26 14:36:04 · 2987 阅读 · 0 评论 -
Couldn‘t build proto file into descriptor pool!
今天使用pytorch-tensorboard的时候出现了这个错误解决方案:出现该问题的原因可能为:1、tensorboard与tensorflow的版本不合适2、tensorboard与tensorboardX的版本不合适4、protobuf的版本不合适尤其是在你的环境中,同时具有tensorboard和tensorboardX时,更有可能出现上述问题。我给出了我所使用的各个版本,在该版本下,可以成功解决该问题:tensorflow:1.13.1tensorboard: 1.原创 2021-03-25 09:50:15 · 2164 阅读 · 0 评论 -
Pytorch Conv1d()
Conv1d一:参数介绍class torch.nn.Conv1d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)in_channels(int) – 输入信号的通道。在文本分类中,即为词向量的维度 out_channels(int) – 卷积产生的通道。有多少个out_channels,就需要多少个1维卷积,实际上就是卷积核的数目。 kernel_si原创 2021-01-20 20:16:22 · 1631 阅读 · 0 评论 -
Pytorch对多维Tensor按维度操作
记忆要点:dim = 0 说明是处理行dim = 1 说明是处理列keepdim = True 保留处理的行/列的特征keepdim = False 不保留处理的行/列的特征网上流传的版本有很多,但是我们根据结果来说话。我的理解是哪个维度发生了变化就是处理的是哪个维度。if __name__ == "__main__": #模型参数初始化 num_input = 784 num_output = 10 W = torch.tensor(np.ra.原创 2020-11-23 20:23:42 · 1973 阅读 · 0 评论 -
pytorch广播语义(Broadcasting semantics)
一般语义(做加减法,但是不影响源张量)如果遵守以下规则,则两个张量是“可播放的”:每个张量至少有一个维度。 两个张量维度不一致的时候,维度小的那个张量直接在最前面自动加一个维度,其余的维度必须保持相等或者其中有一个1 维度即便相等,在做加法减法的时候也需要保持所有维度要么相等要么其中存在一个1 最后生成的维度就是两个张量各自最大的那个维度例如: x = torch.empty(5, 3, 1, 1) y = torch.empty(3, 3, 1) print((x+原创 2020-11-22 16:38:12 · 186 阅读 · 0 评论 -
Pytorch使用多线程加载数据报错
问题: 使用num_workers的时候出现这个问题解决: 将有num_workers的那段代码放到 if __name__ == "__main__":里面原创 2020-11-22 11:27:08 · 1266 阅读 · 3 评论 -
pytorch之手写数字识别
在PyTorch中有两个核心的包,分别是torch和torchvision。在前面的文章中介绍了torch包的一些情况,现在介绍一下torchvision包的一些情况:主要功能是实现数据的处理、导入和预览等,主要处理计算机视觉相关问题的处理工作。下面进入手写数字识别的主要介绍:1.从 torchvison 中导入两个子包 datasets和transformsimport torch...原创 2020-05-10 11:02:08 · 450 阅读 · 0 评论 -
PyTorch之torch.optim
在PyTorch的torch.optim包中提供了非常多的可实现参数自动优化的优化器,比如SGD、AdaGrad、RMSProp、Adam等, 这些类都可以被直接调用,使用起来也非常方便。具体的调用为:一:torch.optim.对应优化器optimizer = torch.optim.Adam(models.parameters(),lr = learning_rate)mo...原创 2020-04-26 16:15:15 · 282 阅读 · 0 评论 -
pytorch之torch.nn
PyTorch中的 torch.nn包提供了很多与实现神经网络中的具体功能相 关的类,这些类涵盖了深度神经网络模型在搭建和参数优化过程中的常 用内容,比如神经网络中的卷积层、池化层、全连接层这类层次构造的 方法、防止过拟合的参数归一化方法、Dropout 方法,还有激活函数部 分的线性激活函数、非线性激活函数相关的方法,等等。首先先看一段代码:import torchfrom torc...原创 2020-04-25 22:56:13 · 1007 阅读 · 0 评论 -
pytorch基础操作
取整函数 torch.abs(变量)加法函数 torch.add(变量,变量)(3)裁剪函数 torch.clamp (变量,上边界,下边界): 输入参数一共有三个,分别是需要进 行裁剪的Tensor数据类型的变量、裁剪的上边界和裁剪的下边界。如果需要裁剪的数小于下边界的话,就将其变为下边界,如果大于上边界的话,就变为上边界,处于边界内的数就直接输入对应的数。(4)...原创 2020-04-25 21:36:19 · 1073 阅读 · 0 评论 -
使用pytorch搭建一个简易神经网络
一.包的引入:import torchbatch_n = 100 #每次迭代个数input_data = 1000 #输入特征数 hidden_layer = 100 #第一个隐层之后的特征数output_data = 10 #输出的变量个数1.Batch_n表示每个批次输入的数据量2.Hidden_layer 表示第一个一层的隐层结束后剩余的特征数3.Inpu...原创 2020-04-25 21:33:23 · 272 阅读 · 0 评论 -
如何查看CUDA版本
之前下载过CUDA,但是没有使用过它,最近突然需要使用到它,但是忘记了版本如何查看。具体步骤:1.按下ctrl+r输入控制面板2.将查看方式点击为小图标3.点击NVIDIA控制面进入NVIDIA4.点击左下角系统信息,进入之后点击组件。对应的NVCUDA.DLL就是对应的版本了...原创 2020-04-22 10:17:34 · 14154 阅读 · 1 评论