********
epoch: 1 time: 8
train_loss: tensor(1.4634, device='cuda:0')
********
epoch: 2 time: 16
train_loss: tensor(1.0485, device='cuda:0')
Finish!