![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
纵心似水
心之所向,素履以往;学如逆旅,一苇以航。
展开
-
提高pytorch在GPU上训练、测试效率的一些实用经验
提高pytorch在GPU上训练、测试效率的一些实用经验1、测试时out of memory可能是因为未设置no_grad:with torch.no_grad(): for data, label in test_loader:2、用watch nvidia-smi命令查看GPU利用率时发现利用率一会儿为100%,一会儿为0%,高低跳动不定,可以在dataloader中设置num...原创 2019-11-21 09:57:27 · 2498 阅读 · 0 评论 -
对抗变分贝叶斯:变分自编码器与生成对抗网络的统一 (一)
最近在学习VAE(变分自编码器), 在github上找到了有关VAE、GAN的许多代码,github链接如下:https://github.com/wiseodd/generative-models我试了试在VAE/adversarial_vb/文件夹中的 AVB代码,并找到了对应的论文Adversarial Variational Bayes: Unifying Variational Au...原创 2019-08-13 20:18:48 · 1586 阅读 · 2 评论 -
对抗变分贝叶斯:变分自编码器与生成对抗网络的统一(二)
文章目录==加载一些必要的库====根据自己制作的图像列表的txt文档加载自己的数据集====定义训练迭代器和测试迭代器====原github代码部分====开始训练迭代====代码运行结果====一点小问题==在上一篇博文也就是对抗变分自编码器——AVB(Adversarial Variational Bayes: Unifying Variational Autoencoders and …...原创 2019-08-14 19:26:36 · 1236 阅读 · 1 评论 -
总结一下今日使用pytorch遇到的几个小问题
这里写自定义目录标题欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题,有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能,丰富你的文章UML 图表FLowchart流程图导出与导入导出导入欢迎使用Ma...原创 2019-08-12 20:05:09 · 6915 阅读 · 7 评论 -
AttributeError: module 'torch.utils' has no attribute 'data'
AttributeError: module ‘torch.utils’ has no attribute ‘data’今天运行pytorch时,突然出现了这么一个错误,可以说原理上不应该出现这个错误,后来在网上找到了原因并进行了修改,不再报错。报错位置:class MyDataset(torch.utils.data.Dataset):参考回答:Fix AttributeError:...原创 2019-08-22 14:31:48 · 3808 阅读 · 2 评论 -
RuntimeError: The size of tensor a (128) must match the size of tensor b (32) at non-singleton dimen
RuntimeError: The size of tensor a (128) must match the size of tensor b (32) at non-singleton dimension 0D_loss = -torch.mean(torch.log(D_real_gauss + TINY) + torch.log(1 - D_fake_gauss + TINY))Run...原创 2019-08-22 14:52:33 · 30593 阅读 · 35 评论 -
实验并解读github上三个DeepDream的Pytorch代码
实验并解读github上三个DeepDream的Pytorch代码今天在学习DeepDream的有关内容,关于论文的翻译已经在启发主义——深入神经网络(Inceptionism: Going Deeper into Neural Networks)一文中进行,而关于代码的实践则在这篇博文中展开。google官方个给出的开源代码是基于Caffe框架的ipynb文件,但在我的环境中运行Caffe必...原创 2019-09-03 20:02:59 · 1278 阅读 · 1 评论 -
Expected type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 ‘weight'
Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 ‘weight’原因:模型和数据未全都放入GPU问题解决:RuntimeError: Expected object of type torch.FloatTensor but found type to...原创 2019-09-11 11:17:56 · 520 阅读 · 0 评论 -
ResNet 18 网络结构
import torchfrom torchvision import modelsresnet = models.resnet18(pretrained=True)print(resnet)"""ResNet( (conv1): Conv2d(3, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False...原创 2019-09-25 17:26:05 · 17244 阅读 · 2 评论