Run in AWS t2.2xlarge:
Step 0
loss 00.503087
[ CPUFloatType{} ]
run cost times: 20, duration: 1508seconds
Step 1
loss 10.0161174
[ CPUFloatType{} ]
run cost times: 20, duration: 1504seconds
valid loss 0.00212624
[ CPUFloatType{} ]