torchrec
torchrec copied to clipboard
Add num_missing_labels, sum_weights and num_positive_samples
Summary: Adds three basic metrics to TorchRec for tracking training statistics:
- NumPositiveSamplesMetric - Weighted sum of positive labels
- SumWeightsMetric - Total sum of sample weights
- NumMissingLabelsMetric - Count of samples with NaN labels
Differential Revision: D87682101
@iamzainhuda has exported this pull request. If you are a Meta employee, you can view the originating Diff in D87682101.