diff --git a/mlperf_logging/rcp_checker/rcp_checker.py b/mlperf_logging/rcp_checker/rcp_checker.py index 8a828f6..065940a 100644 --- a/mlperf_logging/rcp_checker/rcp_checker.py +++ b/mlperf_logging/rcp_checker/rcp_checker.py @@ -268,7 +268,7 @@ def compute_rcp_stats(self): epoch_list.sort() samples_rejected = 4 if record_contents['Benchmark'] == 'unet3d' else 1 record_contents['RCP Mean'] = np.mean(epoch_list[samples_rejected:len(epoch_list)-samples_rejected]) - record_contents['RCP Stdev'] = np.std(epoch_list[samples_rejected:len(epoch_list)-samples_rejected]) + record_contents['RCP Stdev'] = np.std(epoch_list) min_epochs = self._find_min_acceptable_mean( record_contents['RCP Mean'], record_contents['RCP Stdev'],