step = 100, lr = 0.0001, loss = 0.602357
-----------------------------------------------------
step = 200, lr = 0.0001, loss = 0.534716
-----------------------------------------------------