提交 03f4b1d4 编写于 作者: W wangyang59

minor changes on deconv per luotao1 comments

上级 7a322df0
...@@ -353,7 +353,6 @@ TEST(Layer, convTransLayer) { ...@@ -353,7 +353,6 @@ TEST(Layer, convTransLayer) {
testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ false); testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ false);
#ifndef PADDLE_ONLY_CPU #ifndef PADDLE_ONLY_CPU
testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ true); testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ true);
// testConvLayer("cudnn_conv", /* trans= */ false, /* useGpu= */ true);
#endif #endif
} }
......
...@@ -1619,10 +1619,7 @@ def img_conv_layer(input, filter_size, num_filters, ...@@ -1619,10 +1619,7 @@ def img_conv_layer(input, filter_size, num_filters,
param_attr.attr["initial_strategy"] = 0 param_attr.attr["initial_strategy"] = 0
param_attr.attr["initial_smart"] = False param_attr.attr["initial_smart"] = False
if trans: lt = LayerType.CONVTRANS_LAYER if trans else LayerType.CONV_LAYER
lt = LayerType.CONVTRANS_LAYER
else:
lt = LayerType.CONV_LAYER
Layer( Layer(
name=name, name=name,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册