change learning rate and num layers

change learning rate and num layers

First commit

initial commit