PyTorch

MMClassificatio 框架下 Pytorch模型转TensorRT

模型的加载import torchvision.models as modelsresnet34 = models.resnet34()resnet34.load_state_dict(torch.load('latest.pth')['model'])要解决的疑问load_state_dic...
2021-11-14 18:01 阅读数 189

2021-11-13

DAY 3-4 完结1.一些简单损失函数的调用损失函数loss的作用1.计算实际输出和目标之间的差距2.为我们更新输出提供一定的依据（反向传播）, gradimport torchfrom torch.nn import L1Lossfrom torch import nn# 损失函数loss...
2021-11-13 10:53 阅读数 138

MySQL随机函数的运用场景示例总结

1、随机数字-- 随机 0-1select floor(rand() * 2);-- 随机1-100 select floor(rand()*100); -- 随机八位数 select floor(rand()*90000000 + 10000000); -- 随机2位小数select con...
2021-11-12 21:34 阅读数 60

使用Pytorch构建一个神经网络

关于torch.nn·使用Pytorch来构建神经网络，主要的工具都在torch.nn包中·nn依赖于autograd来定义模型，并对其自动求导构建神经网络的典型流程·定义一个拥有可学习参数的神经网络·遍历训练数据集·处理输入数据使其流经神经网络·计算损失值...
2021-10-24 00:56 阅读数 51

torch.nn神经网络（2）--池化层+非线性激活函数（ReLU和Sigmiod）+Sequential()函数的使用

nn.Module神经网络4.池化层池化层：池化函数使用某一位置的相邻输出的总体统计特征来代替网络在该位置的输出。 本质是降采样，减少网络的参数量还是老规矩，导入模块及数据集，CIFAR10,batchsize=64：import torchimport torchvisionfrom tor...
2021-10-23 00:00 阅读数 48

【深度学习】 loss不下降，准确率很低的原因

loss不下降，ACC很低（只有0.1，0.2这种）可能的原因有：数据集有问题（噪声过多或存在过多的标签错误或类别不平衡）梯度爆炸梯度消失笔者遇到的梯度爆炸情况下图的矩阵是pooler_output（从bert得到的句子向量）：若干个不同的文本，在训练两个batch后可见模型的输出几乎一样了，...
2021-09-29 14:41 阅读数 68

关于深度学习中使用多卡GPU进行训练学习总结

1.nn.parallel.scatter 向多个设备分发参数这是实现方法，在模型中可以值调用函数，其原理就是通过for循环 然后copy到不同的设备上2.allreduce 函数 将所有向量相加，并将结果广播给所有的gpu
2021-09-13 17:25 阅读数 62

基于Pytorch的torch.nn.embedding()实现词嵌入层

 nn.embedding()其实是NLP中常用的词嵌入层，在实现词嵌入的过程中embedding层的权重用于随机初始化词的向量，该embedding层的权重参数在后续训练时会不断更新调整，并被优化。  nn.embedding:这是一个矩阵类，该开始时里面初始化了一个随机矩阵，矩阵的长是字典...
2021-07-28 15:21 阅读数 351

深度学习实战——基于pytorch写的一个猫狗分类的模型——学习总结

写在前面本学习总结主要目的1.怕以后忘记，到时候翻来看看，2.梳理一下项目过程，便于工业化生产（哈哈）3.敲一遍代码加深记忆4.帮助他人，方便入门同学能够容易入门本过程主要设计到的内容1.torch.nn，nn模块下的Module类，组件类，neture network2.tensor，巩固t...
2021-07-23 17:07 阅读数 71

深度学习之Pytorch实现房价预测

废话不多说准备数据# 为实验准备数据def init_data(): x = torch.FloatTensor(torch.linspace(0, 100)) y = x + torch.FloatTensor(torch.randn(100)) * 10 return x, y...
2021-06-22 11:50 阅读数 78
技术
今日推荐
ioDraw Mac客户端安装教程
阅读数 19
下载桌面版
GitHub
百度网盘（提取码：draw）
Gitee
云服务器优惠
阿里云优惠券
腾讯云优惠券
华为云优惠券
站点信息
问题反馈
邮箱：[email protected]
QQ群：766591547
关注微信
©2020-2024 ioDraw All rights reserved, Privacy Policy