Skip to content

Commit 0137757

Browse files
authored
Fix get_batch_loss_metrics comments (#413)
## Summary Remove misleading docstring in `get_batch_loss_metrics()` of `test/utils.py`. <!--- ## Details This is an optional section; is there anything specific that reviewers should be aware of? ---> ## Testing Done <!--- This is a required section; please describe how this change was tested. ---> <!-- Replace BLANK with your device type. For example, A100-80G-PCIe Complete the following tasks before sending your PR, and replace `[ ]` with `[x]` to indicate you have done them. --> - Hardware Type: A10G - [ ] run `make test` to ensure correctness - [X] run `make checkstyle` to ensure code style - [ ] run `make test-convergence` to ensure convergence Signed-off-by: Austin Liu <[email protected]>
1 parent 7e3683e commit 0137757

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -483,7 +483,7 @@ def get_batch_loss_metrics(
483483
ref_bias: torch.FloatTensor = None,
484484
average_log_prob: bool = True,
485485
):
486-
"""Compute the ORPO loss and other metrics for the given batch of inputs for train or test."""
486+
"""Compute the loss metrics for the given batch of inputs for train or test."""
487487

488488
forward_output = self.concatenated_forward(
489489
_input, weight, target, bias, average_log_prob

0 commit comments

Comments
 (0)