Lines Matching full:loss

31             # calculate loss
33                 loss = torch.zeros(1, device=device)
35                     loss = loss + criterion(target, y.squeeze(1))
36                 loss = loss / len(output)
38                 loss = criterion(target, output.squeeze(1))
41             loss.backward()
53             # update running loss
54             running_loss += float(loss.cpu())
89                 # calculate loss
90                 loss = criterion(target, output.squeeze(1))
92                 # update running loss
93                 running_loss += float(loss.cpu())