Hi everybody, well I’m a little dubious about the result I got.
- Is it reasonable to have these figures for the first epoch?
- which Loss formula should I take to draw the LOSS_EPOCH graph? (Loss_sum, loss, or epoch_loss??)
for i, (xA, xB, score) in enumerate(train_loader, 1):
....
loss = criterion(..)
total_loss += loss.item()
total_loss2 += loss.item() * len(xA)
if i%5==0:
Loss_sum = ( total_loss / i ) * len(xA) ## Loss_sum(total) in code below
loss= total_loss / i ## loss(mean) in code below
#PRINT
.
.
epoch_loss = total_loss2 / len(train_loader.dataset)
....
#for evaluation
total_loss_eval+= loss.item()
cum_loss += (loss.item()*len(XA))
if i%3 ==0:
print(f'total_loss_eval{total_loss_eval / i * len(XA)}')
LOSS_eval= cum_loss / len(eval_loader2.dataset) ### EVAl Epoch in result below
Herein lies the first epoch output.
[0m 15s] Train Epoch: 1 [250/4500 (6%)] Loss_sum(total): 5.72 loss(mean) 0.11438
[0m 30s] Train Epoch: 1 [500/4500 (11%)] Loss_sum(total): 6.32 loss(mean) 0.12634
[0m 46s] Train Epoch: 1 [750/4500 (17%)] Loss_sum(total ): 6.49 loss(mean) 0.12977
begin validation ...
total_loss_eval 6.013869825336668
EVAl Epoch: 1 [500] Loss: 0.12
save currently the best model to [/content/model.pt]
[1m 3s] Train Epoch: 1 [1000/4500 (22%)] Loss_sum(total): 6.67 loss(mean) 0.13343
[1m 19s] Train Epoch: 1 [1250/4500 (28%)] Loss_sum(total): 6.29 loss(mean) 0.12571
.
.
total_loss_eval 2.934242847065131
EVAl Epoch: 1 [500] Loss: 0.06
.
.
[4m 43s] Train Epoch: 1 [4250/4500 (94%)] Loss_sum(total): 3.65 loss(mean) 0.07292
[4m 58s] Train Epoch: 1 [4500/4500 (100%)] Loss_sum(total): 3.56 loss(mean) 0.07124
begin validation ...
total_loss_eval 2.032051028476821
EVAl Epoch: 1 [500] Loss: 0.04
epoch_loss 0.0712374652425448
..........