pytorch
起名字什么的好难
在自我提升的路上奔腾不息~
展开
-
pytorch nn.conv2d参数个数计算
nn.conv2d(256, 256, 3, 3)假设输入大小为[3, 128, 128],输出的大小则为128-3+1=126,这一层参数个数为256*3*3+256=2560.假设输入尺寸为[C_in, H, W],则输出大小计算公式为[C_out, (H-K+2*P)/S+1, (W-K+2*P)/S+1]。其中C_in是输入通道数,H,W是数据的尺寸,K是核大小,P是padding大小,S是步长stride大小。nn.conv2d(C_in, C_out, K, K)参数个数的计算公式为C_原创 2021-03-12 18:23:00 · 4060 阅读 · 0 评论 -
ImportError: IProgress not found. Please update jupyter and ipywidgets.
解决方法:conda install -n base -c conda-forge widgetsnbextensionconda install -n py36 -c conda-forge ipywidgets注意:-n指定了anaconda创建的环境名。第一条语句-n base指的是anaconda的基本环境,第二条语句的-n py36需要根据具体情况设定。像我的环境是torch1.4.0,相应的我第二句命令就是conda install -n torch1.4.0 -c conda-forg原创 2021-03-11 11:30:58 · 817 阅读 · 0 评论 -
pytorch网络模型结构的总结打印
在keras中可以通过model.summary()打印出模型的结构,类似这样:在pytorch中想要实现类似的功能,直接打印模型就可以了。例如from torchvision import modelsmodel = models.vgg16()print(model)输出结果VGG ( (features): Sequential ( (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))原创 2021-03-03 21:09:00 · 6703 阅读 · 1 评论 -
RuntimeError: Found 0 images in subfolders of: ./data
在使用torchvision.datasets.ImageFolder(path, transform)进行加载图片时报错。出错代码如下。dataset = torchvision.datasets.ImageFolder('../cocotrain/', transform=data_transform)data_loader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=True)报的错误是:Runt原创 2021-03-03 21:04:10 · 1109 阅读 · 1 评论 -
PyTorch中使用Tensorboard
内容出自A Complete Guide to Using TensorBoard with PyTorch。建议使用jupyter notebook或者jupyter lab进行代码运行,可以执行每一个代码段,并看到结果。1. conda安装tensorboardconda install -c conda-forge tensorboard也可以使用pippip install tensorboard2. 导入一些包其中直接和使用tensorboard相关的是from torch.uti原创 2021-01-14 15:23:55 · 2271 阅读 · 0 评论