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