Pytorch
会飞的糖企鹅
这个作者很懒,什么都没留下…
展开
-
Pytorch日志3-模型加速方法与技巧
方法一:量化模型变量的方式 (1)采用GPU训练 (2)用pytorch的新版本对模型进行量化,把32位的数据结构将为8位。 (3)在测试与使用的使用,采用CPU跑前向通道。 方法二:待补充 ...原创 2019-12-10 10:15:58 · 119 阅读 · 0 评论 -
Pytorch日志2-对于pytorch下载数据集慢的解决方法
Pytorch日志2-对于pytorch下载数据集慢的解决方法 这个问题很简单,只需要在项目下面新建一个data文件夹,然后将自己事先下载好的数据集放进去就好了,也不需要解压。 ...原创 2019-11-21 15:33:45 · 937 阅读 · 0 评论 -
Pytorch日志1-固定一部分模型,仅训练全链接层的参数
Pytorch日志1-固定一部分模型,仅训练全连接层的参数 (1)只训练全连接层参数 // An highlighted block # 选择模型 model = models.vgg19(pretrained=True) # 我们选择预训练好的模型vgg19 for parma in model.parameters(): parma.requires_grad = False m...原创 2019-11-20 15:38:12 · 1306 阅读 · 0 评论 -
数学基本知识2-卷积输入输出通道计算和如何绘制网络的tensorboardx图
数学基本知识2-卷积输入输出通道计算 卷积的计算,涉及输入通道,输出通道,步长,卷积核大小等,在神经网络的计算中,计算输入输出的通道是设计网络的关键。本文主要介绍了卷积输入输出的主要方法和采用tensorboardx查看网络模型的介绍。 一、nn.Conv1d 一维卷积计算 (1)一维的卷积能处理多维数据 nn.Conv2d(self, in_channels, out_channels, ker...原创 2019-11-18 16:55:53 · 680 阅读 · 0 评论 -
Ubuntu18.04LTS日志14-Pytorch编程常用功能
Ubuntu18.04LTS日志14-Pytorch编程常用功能 (1)选择指定GPU跑程序 // choose your GPU number import os os.environ['CUDA_VISIBLE_DEVICES'] = '1'原创 2019-11-13 14:46:46 · 113 阅读 · 0 评论 -
Ubuntu18.04LTS日志10-pytorch训练模型与保存
Ubuntu18.04LTS日志10-pytorch训练模型与保存 参考链接: https://blog.csdn.net/sll71/article/details/95856952原创 2019-11-12 14:45:50 · 324 阅读 · 0 评论