diff --git a/scripts/benchmarks/benchmark_utils.py b/scripts/benchmarks/benchmark_utils.py index 71e450402a..c30f82a54b 100644 --- a/scripts/benchmarks/benchmark_utils.py +++ b/scripts/benchmarks/benchmark_utils.py @@ -853,7 +853,6 @@ def run(self): batch_size, sequence_length) except Exception as e: - raise e train_time = np.nan train_memory = np.nan train_result[model_name][workload] = (train_time, train_memory)