multi layers issue

[kelizhong]

Hi,

I try to set the num_layers value in main.y to support multi layers, but I got below error: assert arg_shapes is not None, "shape inference failed" AssertionError: shape inference failed

Do you have any suggestion about it