step = 100, lr = 0.0001, loss = 0.569343
-----------------------------------------------------
step = 200, lr = 0.0001, loss = 0.493275
-----------------------------------------------------
step = 100, lr = 0.0001, loss = 0.631249
-----------------------------------------------------
step = 200, lr = 0.0001, loss = 0.568028
-----------------------------------------------------