from local

This commit is contained in:
dohe0342 2023-01-09 20:26:20 +09:00
parent 1e29fe0719
commit ac8ae31cae
2 changed files with 0 additions and 2 deletions

View File

@ -714,8 +714,6 @@ def compute_loss(
# Note: We use reduction=sum while computing the loss.
info["loss"] = loss.detach().cpu().item()
info["simple_loss"] = simple_loss.detach().cpu().item()
info["pruned_loss"] = pruned_loss.detach().cpu().item()
return loss, info