quangnhien

Loss function

2

 I see that you have multiple nllLoss with train_size. This makes the value of the loss is very large. I don't understand why do you do that? You can...