提交 5fff96f5 编写于 作者: W wangyang59

add another small test in test_LayerGrad for convTransLayer

上级 e68b50ad
...@@ -345,6 +345,8 @@ void testConvTransLayer(const string& type, bool trans, bool useGpu) { ...@@ -345,6 +345,8 @@ void testConvTransLayer(const string& type, bool trans, bool useGpu) {
config.layerConfig.num_filters()); config.layerConfig.num_filters());
testLayerGrad(config, "convTrans", 100, trans, useGpu); testLayerGrad(config, "convTrans", 100, trans, useGpu);
// Use small batch_size and useWeight=true to test biasGrad
testLayerGrad(config, "convTrans", 2, trans, useGpu, true, 0.02);
} }
TEST(Layer, convTransLayer) { TEST(Layer, convTransLayer) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册