Fix typo in loss message

4b1c4488 · Gustaf Ahdritz · 1feca912 · 4b1c4488
Commit 4b1c4488 authored Nov 03, 2021 by Gustaf Ahdritz
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

openfold/utils/loss.py openfold/utils/loss.py +1 -1

No files found.
--- a/openfold/utils/loss.py
+++ b/openfold/utils/loss.py
@@ -1496,7 +1496,7 @@ class AlphaFoldLoss(nn.Module):
            if weight:
                loss = loss_fn()
                if(torch.isnan(loss) or torch.isinf(loss)):
-                    logging.warning(f"{k} loss is NaN. Skipping example...")
+                    logging.warning(f"{loss_name} loss is NaN. Skipping...")
                    loss = loss.new_tensor(0., requires_grad=True)
                cum_loss = cum_loss + weight * loss