![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch学习笔记
鱼木木和木木鱼
这个作者很懒,什么都没留下…
展开
-
torch.nn.functional.conv2d的用法
torch.nn.functional.conv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1) → Tensor注意:!!!对F.conv2d一样适用翻译 2020-03-01 17:59:49 · 5134 阅读 · 0 评论 -
pytorch中训练模型的保存和提取
有两种方式只保存模型的参数:torch.save(the_model.state_dict(), PATH)、之后需要的时候把模型参数再次提取出来:the_model = TheModelClass(*args, **kwargs)the_model.load_state_dict(torch.load(PATH))保存整个模型torch.save(the_model, PAT...翻译 2020-02-27 21:49:14 · 681 阅读 · 0 评论 -
pytorch中BatchNorm2d的用法
CLASS torch.nn.BatchNorm2d(num_features, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)基本原理为:如图所示,块标准化的目的就是让传输的数据合理的分布,加速训练的过程。输入为一个四维数据(N,C,H,W),N-输入的batch size,C是输入的图像的通道数,(H,W...翻译 2020-02-26 20:09:39 · 4854 阅读 · 0 评论 -
torch.utils.data.DataLoader的用法
torch.utils.data.DataLoader翻译 2020-02-24 20:32:50 · 2932 阅读 · 0 评论 -
TORCHVISION.TRANSFORMS包的用法
TORCHVISION.TRANSFORMS包中部分类的用法CLASS torchvision.transforms.Compose(transforms)将需要进行变换的操作组合起来 例如```python>>> transforms.Compose([>>> transforms.CenterCrop(10),>>>...翻译 2020-02-23 18:12:16 · 204 阅读 · 0 评论