search

ry / tensorflow-resnet

ResNet model in TensorFlow
MIT License
1.66k stars 625 forks source link

block function in resnet.py #37

Open ljbthu opened 6 years ago

ljbthu commented 6 years ago

I think that there is a small mistake when defining the 'bottleneck' structure in the block function . In your code, three 1x1 conv layers ('a', 'b', 'c') are used to constitute the 'bottleneck' block. I believe you forgot c['ksize'] = 3 when defining layer 'b'.

liyongze commented 5 years ago

I think so. c['ksize'] should be definited 3 here.