Open joey3639570 opened 6 years ago
In some caffe version, they use the BatchNorm layer and a Scale layer to implement the BN, And I think in caffe2, the BN layer just rely on the spatial_bn which is shown in https://github.com/caffe2/caffe2/blob/929c33cee0942e05dd36be465ad04e7625f3715f/caffe2/python/models/resnet.py so just add the spaital_bn!
@JiYuanFeng Thank you so much! This helps me a lot! I think that all I need to do now is to solve the problem of element wise layer. Thank you again! Btw, It seems that I can't use your link, could you post it again?
I'm new to caffe2 and deep learning. I was building my resnet152 model, and I was stuck on building "scale layer" and "element wise layer". Here's part of my codes.
The first question is how to build scale layer. I found out that I can't use brew to build scale layer. What should I do or import to build a scale layer?
The second question is for the elementwise layer. Although I know I can use brew to build it, I don't know what to type in. I hope that someone can show me an example of using them.