add another small test in test_LayerGrad for convTransLayer

avx_docs
wangyang59 9 years ago
parent e68b50ad09
commit 5fff96f532

@ -345,6 +345,8 @@ void testConvTransLayer(const string& type, bool trans, bool useGpu) {
config.layerConfig.num_filters());
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) {

Loading…
Cancel
Save