2020
10-08
10-08
PyTorch之nn.ReLU与F.ReLU的区别介绍
我就废话不多说了,大家还是直接看代码吧~importtorch.nnasnnimporttorch.nn.functionalasFimporttorch.nnasnnclassAlexNet_1(nn.Module):def__init__(self,num_classes=n):super(AlexNet,self).__init__()self.features=nn.Sequential(nn.Conv2d(3,64,kernel_size=3,stride=2,padding=1),nn.BatchNorm2d(64),nn.ReLU(inplace=True),)defforward(self,x):x=self...
继续阅读 >