minor changes on deconv per luotao1 comments

avx_docs
wangyang59 9 years ago
parent 7a322df0a8
commit 03f4b1d4d2

@ -353,7 +353,6 @@ TEST(Layer, convTransLayer) {
testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ false);
#ifndef PADDLE_ONLY_CPU
testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ true);
// testConvLayer("cudnn_conv", /* trans= */ false, /* useGpu= */ true);
#endif
}

@ -1619,10 +1619,7 @@ def img_conv_layer(input, filter_size, num_filters,
param_attr.attr["initial_strategy"] = 0
param_attr.attr["initial_smart"] = False
if trans:
lt = LayerType.CONVTRANS_LAYER
else:
lt = LayerType.CONV_LAYER
lt = LayerType.CONVTRANS_LAYER if trans else LayerType.CONV_LAYER
Layer(
name=name,

Loading…
Cancel
Save