![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
信号傅老师
这个作者很懒,什么都没留下…
展开
-
Pytorch中 list(net.parameters())[0]
Pytorch中 list(net.parameters())[0]先看一下Lenet的代码class LeNet(nn.Module): def __init__(self): super().__init__() self.conv = nn.Sequential( nn.Conv2d(1, 6, 5), # in_chan...原创 2019-12-16 21:44:01 · 2484 阅读 · 0 评论 -
学习pytorch
1、print('cuda') if torch.cuda.is_available() else print('cpu')输出cuda这就像if else语句一样2、super(LeNet, self).__init__()这是python2中的写法,python3中在子类的初始化函数中,关联父类的初始化方法,不用再super()函数中传参super().__init__()...原创 2019-12-13 10:46:10 · 152 阅读 · 0 评论 -
BN层pytorch实现
# Created by Xky at 2019/11/29import timeimport torchimport torchvisionimport torch.nn as nnimport sysimport torchvision.transforms as transformsfrom torch.utils.data.dataloader import DataLoad...原创 2019-11-29 14:12:32 · 3237 阅读 · 0 评论