将层进行编码,相当于进入nn.sequential,但是立刻就会出现上述错误,data进入之后,并且训练不起来,loss值不变
”部分代码如下,这里只针对错误“
def make_layer(self, type, inchannels, outchannels, RU_amount,
growth_rate, num_block, max_input_channels , pool_type):
layers = []
if type == 1:
layers.append(ResNetUnit(RU_amount, inchannels, outchannels))
elif type == 2