diff --git a/paddle/gserver/layers/SwitchOrderLayer.cpp b/paddle/gserver/layers/SwitchOrderLayer.cpp index 6a91042f628920a9986763531fb4c633307b43b8..5b2cfe3c48d65d85d792d8817afd444b00561104 100644 --- a/paddle/gserver/layers/SwitchOrderLayer.cpp +++ b/paddle/gserver/layers/SwitchOrderLayer.cpp @@ -32,10 +32,10 @@ bool SwitchOrderLayer::init(const LayerMap& layerMap, outDims_ = TensorShape(4); auto& reshape_conf = config_.reshape_conf(); - for (size_t i = 0; i < reshape_conf.heightaxis_size(); i++) { + for (int i = 0; i < reshape_conf.heightaxis_size(); i++) { heightAxis_.push_back(reshape_conf.heightaxis(i)); } - for (size_t i = 0; i < reshape_conf.widthaxis_size(); i++) { + for (int i = 0; i < reshape_conf.widthaxis_size(); i++) { widthAxis_.push_back(reshape_conf.widthaxis(i)); } createFunction(nchw2nhwc_, "NCHW2NHWC", FuncConfig());