Skip to content

Commit 7f581e3

Browse files
authored
Merge pull request #377 from mlcommons/fix_rcp_stdev
Calculate Stdev without olympic pruning
2 parents 369260b + 52245a0 commit 7f581e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlperf_logging/rcp_checker/rcp_checker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ def compute_rcp_stats(self):
268268
epoch_list.sort()
269269
samples_rejected = 4 if record_contents['Benchmark'] == 'unet3d' else 1
270270
record_contents['RCP Mean'] = np.mean(epoch_list[samples_rejected:len(epoch_list)-samples_rejected])
271-
record_contents['RCP Stdev'] = np.std(epoch_list[samples_rejected:len(epoch_list)-samples_rejected])
271+
record_contents['RCP Stdev'] = np.std(epoch_list)
272272
min_epochs = self._find_min_acceptable_mean(
273273
record_contents['RCP Mean'],
274274
record_contents['RCP Stdev'],

0 commit comments

Comments
 (0)