是的,在Residual这个类中增加两层网络,和现有的网络一样
时间: 2024-02-22 22:58:06 浏览: 70
经典CNN网络之ResNet101 对厨余垃圾8分类数据集的分类任务
好的,那我们可以在Residual类中增加两层全连接层(Linear层),代码如下:
```
class Residual(Module):
def __init__(self, input, out):
super(Residual, self).__init__()
self.fc1 = Linear(input, out)
self.bn = BatchNorm1d(out)
self.relu = ReLU()
self.fc2 = Linear(out, out)
self.fc3 = Linear(out, out)
def forward(self, input):
out = self.fc1(input)
out = self.bn(out)
out = self.relu(out)
out = self.fc2(out)
out = self.bn(out)
out = self.relu(out)
out = self.fc3(out)
return torch.cat([out, input], dim=1)
```
这样,我们在原来的Residual类中增加了两层全连接层,每层都有out个输出节点,输入节点数和输出节点数相同。其中,我们使用了ReLU作为激活函数,BatchNorm1d作为归一化函数。
阅读全文