pytorch实现特殊的Module–Sqeuential三种写法
我就废话不多说了,直接上代码吧! # -*- coding: utf-8 -*- #@Time :2019/7/1 13:34 #@Author :XiaoMa import torch as t from torch import nn #Sequential的三种写法 net1=nn.Sequential() net1.add_module('conv',nn.Conv2d(3,3,3)) #Conv2D(输入通道数,输出通道数,卷积核大小) net1.add_module('batchnorm',nn.BatchNorm2d(3)) #BatchNorm2d(特征数) net1.
下载地址
用户评论