这是一个适合PyTorch入门者看的博客。PyTorch的文档质量比较高，入门较为容易，这篇博客选取[官方链接](http://pytorch.org/tutorials/beginner/transfer_learni…

之前我们使用nn.Sequential()都是直接写死的，就如下所示： # Example of using Sequential model = nn.Sequential( nn.Conv2d(1,20,5), nn…

Y7000P电脑环境i7处理器，1060显卡，16g内存，win10家庭版（系统版本号1809），在联想官网升级过bios，所有驱动都是最新。（截止时间点2019年3月1日） python3.5    …

搭建神经网络模型 pytorch模型的搭建有两种方法，第一种是nn.Module搭配forward 第二种是nn.sequential 贴上一个简单的cnn网络，体会一下 利用nn.Module()搭建简单cnn网络 c…

远程实时loss import visdom vis = visdom.Visdom() try: win = vis.line( Y=np.column_stack(([loss], [valid_loss])), X…

model.eval()，pytorch会自动把BN和DropOut固定住，不会取平均，而是用训练好的值。不然的话，一旦test的batch_size过小，很容易就会被BN层导致生成图片颜色失真极大；在模型测试阶段使用 …

转载请注明出处： https://www.cnblogs.com/darkknightzh/p/11332155.html 代码网址： https://github.com/darkknightzh/TensorRT_p…

昨天发了一篇PyTorch在64位Windows下的编译过程的文章，有朋友觉得能不能发个包，这样就不用折腾了。于是，这个包就诞生了。感谢@Jeremy Zhou为conda包的安装做了测试。 更新：从0.4.0版本开始，…

1 import torch 2 import torch.nn as nn 3 from torch.autograd import Variable 4 5 word2id = {'hello': 0, 'world…

因为ubuntu 系统是14.0的,安装pytorch1.0的时候,本身已经安装好了cuda8,在验证gpu的时候，torch.cuda.is_available()返回false 安装命令是： conda instal…

这两天一直迷惑pytorch中关于requires_grad的设置，特别是在写cuda extension时。今天在pytorch github上看到个issue解释的很不错，姑且记录在这以备后用。 我的问题是在写cud…

RuntimeError: bool value of Tensor with more than one value is ambiguous 运行下面这段代码的时候出错了，后来网上搜说改成 if w1.grad is…