class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.residual_layer = self.make_layer(Conv_ReLU_Block, 18)
self.input = nn.Conv2d(in_channels=1, out_channels=64, kernel_size=3, stride=1, padding=1, bias=False)
self.output = nn.Conv2d(in_channels=64, out_channels=1, kernel_size=3, stride=1, padding=1, bias=False)
self.relu = nn.ReLU(inplace=True)
这里是继承里面需要学习的:当父类没有需要的方法时,子类可以直接定义新方法;当父类中有需要的方法,但是子类同样的方法发生了变化,这个时候就需要改写该方法,直接重新定义该方法即可;特殊情况:子类的方法与父类的方法大体上一致,这时候可以通过super函数调用父类的方法,然后再补充上子类特殊方法即可。