Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
temporarily remove hybridize in test_gluon_rnn.test_layer_fill_shape
Browse files Browse the repository at this point in the history
  • Loading branch information
szha committed Aug 4, 2018
1 parent 785b7a0 commit e631019
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion tests/python/unittest/test_gluon_rnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -547,7 +547,6 @@ def test_cell_fill_shape():
@assert_raises_cudnn_disabled()
def test_layer_fill_shape():
layer = gluon.rnn.LSTM(10)
layer.hybridize()
check_rnn_layer_forward(layer, mx.nd.ones((3, 2, 7)))
print(layer)
assert layer.l0_i2h_weight.shape[1] == 7, layer.l0_i2h_weight.shape[1]
Expand Down

0 comments on commit e631019

Please sign in to comment.