epoch, loss(train)
0, 0.136450
5, 0.068499
10, 0.063806
10, 0.063806