Loss varies with epochs like this.
Loss:2.0544769048690794
Loss:0.6927875459194184
Loss:0.15503601282835006
Loss:0.043137667886912825
Loss:0.020370384491980077
Loss:0.013511684630066157
Loss:0.010588872153311968
Loss:0.008972734585404396
Loss:0.007898936001583933
Loss:0.00709811644628644