提交 fa029636 编写于 作者: W wanghaoshuang

Delete debug log

上级 e8ee07f3
...@@ -33,11 +33,9 @@ bool SwitchOrderLayer::init(const LayerMap& layerMap, ...@@ -33,11 +33,9 @@ bool SwitchOrderLayer::init(const LayerMap& layerMap,
auto& reshape_conf = config_.reshape_conf(); auto& reshape_conf = config_.reshape_conf();
for (size_t i = 0; i < reshape_conf.heightaxis_size(); i++) { for (size_t i = 0; i < reshape_conf.heightaxis_size(); i++) {
LOG(INFO) << "reshape height axis: " << reshape_conf.heightaxis(i);
heightAxis_.push_back(reshape_conf.heightaxis(i)); heightAxis_.push_back(reshape_conf.heightaxis(i));
} }
for (size_t i = 0; i < reshape_conf.widthaxis_size(); i++) { for (size_t i = 0; i < reshape_conf.widthaxis_size(); i++) {
LOG(INFO) << "reshape width axis: " << reshape_conf.widthaxis(i);
widthAxis_.push_back(reshape_conf.widthaxis(i)); widthAxis_.push_back(reshape_conf.widthaxis(i));
} }
createFunction(nchw2nhwc_, "NCHW2NHWC", FuncConfig()); createFunction(nchw2nhwc_, "NCHW2NHWC", FuncConfig());
...@@ -60,8 +58,6 @@ void SwitchOrderLayer::setOutDims() { ...@@ -60,8 +58,6 @@ void SwitchOrderLayer::setOutDims() {
reshapeWidth_ *= outDims_[widthAxis_[i]]; reshapeWidth_ *= outDims_[widthAxis_[i]];
} }
output_.setFrameWidth(reshapeWidth_); output_.setFrameWidth(reshapeWidth_);
LOG(INFO) << "outDims: " << outDims_[0] << "; " << outDims_[1] << ";"
<< outDims_[2] << ";" << outDims_[3];
} }
void SwitchOrderLayer::setInDims() { void SwitchOrderLayer::setInDims() {
...@@ -76,8 +72,6 @@ void SwitchOrderLayer::setInDims() { ...@@ -76,8 +72,6 @@ void SwitchOrderLayer::setInDims() {
int totalCount = input->getElementCnt(); int totalCount = input->getElementCnt();
int channels = totalCount / (inDims_[0] * inDims_[2] * inDims_[3]); int channels = totalCount / (inDims_[0] * inDims_[2] * inDims_[3]);
if (channels != 0) inDims_.setDim(1, channels); if (channels != 0) inDims_.setDim(1, channels);
LOG(INFO) << "inDims: " << inDims_[0] << "; " << inDims_[1] << ";"
<< inDims_[2] << ";" << inDims_[3];
} }
void SwitchOrderLayer::forward(PassType passType) { void SwitchOrderLayer::forward(PassType passType) {
...@@ -95,12 +89,12 @@ void SwitchOrderLayer::forward(PassType passType) { ...@@ -95,12 +89,12 @@ void SwitchOrderLayer::forward(PassType passType) {
inputs.addArg(*getInputValue(0), inDims_); inputs.addArg(*getInputValue(0), inDims_);
outputs.addArg(*getOutputValue(), outDims_); outputs.addArg(*getOutputValue(), outDims_);
nchw2nhwc_[0]->calc(inputs, outputs); nchw2nhwc_[0]->calc(inputs, outputs);
// forwardActivation(); forwardActivation();
} }
void SwitchOrderLayer::backward(const UpdateCallback& callback) { void SwitchOrderLayer::backward(const UpdateCallback& callback) {
(void)callback; (void)callback;
// backwardActivation(); backwardActivation();
// switch NHWC to NCHW // switch NHWC to NCHW
BufferArgs inputs; BufferArgs inputs;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册