Skip to content

Commit

Permalink
Add test for issue #327 (dmlc#328)
Browse files Browse the repository at this point in the history
* Added test for adding down-/up-sampled layers

* Enabled test for adding down-/up-sampled layers

* Normalize whitespace
  • Loading branch information
thefiddler authored and tqchen committed Jan 19, 2018
1 parent 94c8724 commit 3bf7785
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion tests/python/frontend/keras/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def get_tvm_output(x, target, ctx, input_name='data', dtype='float32'):
tvm_out = get_tvm_output(x.transpose([0,3,1,2]), target, ctx)
np.testing.assert_allclose(keras_out, tvm_out, rtol=1e-5, atol=1e-5)


def test_forward_elemwise_add():
print("test_forward_elemwise_add")
r = []
Expand All @@ -58,6 +58,16 @@ def test_forward_elemwise_add():
y = keras.layers.GlobalAveragePooling2D()(y)
keras_model = keras.models.Model(data, y)
verify_keras_frontend(keras_model)


def test_forward_elementwise_add2():
data = keras.layers.Input(shape=(32,32,3))
r = keras.layers.Conv2D(10, (3, 3), padding="same")(data)
x = keras.layers.Conv2D(10, (3, 3), strides=(2, 2), padding="same")(data)
x = keras.layers.UpSampling2D()(x)
x = keras.layers.add([x, r])
x = keras.layers.GlobalAveragePooling2D()(x)
keras_model = keras.models.Model(data, x)


def test_forward_softrelu():
Expand Down Expand Up @@ -136,6 +146,7 @@ def test_forward_resnet50():

if __name__ == '__main__':
test_forward_elemwise_add()
test_forward_elementwise_add2()
test_forward_softrelu()
test_forward_leaky_relu()
test_forward_dense()
Expand Down

0 comments on commit 3bf7785

Please sign in to comment.