diff --git a/src/transformers/benchmark/benchmark_args_utils.py b/src/transformers/benchmark/benchmark_args_utils.py index cb94b70a..49042258 100644 --- a/src/transformers/benchmark/benchmark_args_utils.py +++ b/src/transformers/benchmark/benchmark_args_utils.py @@ -62,8 +62,8 @@ class BenchmarkArguments: fp16: bool = field(default=False, metadata={"help": "Use FP16 to accelerate inference."}) training: bool = field(default=False, metadata={"help": "Benchmark training of model"}) verbose: bool = field(default=False, metadata={"help": "Verbose memory tracing"}) - no_speed: bool = field(default=False, metadata={"help": "Don't perform speed measurments"}) - no_memory: bool = field(default=False, metadata={"help": "Don't perform memory measurments"}) + no_speed: bool = field(default=False, metadata={"help": "Don't perform speed measurements"}) + no_memory: bool = field(default=False, metadata={"help": "Don't perform memory measurements"}) trace_memory_line_by_line: bool = field(default=False, metadata={"help": "Trace memory line by line"}) save_to_csv: bool = field(default=False, metadata={"help": "Save result to a CSV file"}) log_print: bool = field(default=False, metadata={"help": "Save all print statements in a log file"})