def stack_fn(x): x = resnet.stack2(x, 64, 3, name='conv2') x = resnet.stack2(x, 128, 4, name='conv3') x = resnet.stack2(x, 256, 23, name='conv4') return resnet.stack2(x, 512, 3, stride1=1, name='conv5')
def stack_fn(x):
x = resnet.stack2(x, 64, 3, name="conv2")
x = resnet.stack2(x, 128, 4, name="conv3")
x = resnet.stack2(x, 256, 23, name="conv4")
return resnet.stack2(x, 512, 3, stride1=1, name="conv5")