From b1f5955229450454286faf002a8736d7c7e5ad20 Mon Sep 17 00:00:00 2001 From: Xingjian Shi Date: Mon, 10 Aug 2020 11:18:43 -0700 Subject: [PATCH] update --- scripts/benchmarks/benchmark_utils.py | 2 +- scripts/conversion_toolkits/README.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/benchmarks/benchmark_utils.py b/scripts/benchmarks/benchmark_utils.py index a6cb6308e9..0d6f0e0dad 100644 --- a/scripts/benchmarks/benchmark_utils.py +++ b/scripts/benchmarks/benchmark_utils.py @@ -835,7 +835,7 @@ def run(self): infer_time, infer_memory = self.inference_speed_memory(model_name, batch_size, sequence_length) - except Exception: + except Exception as e: infer_time = np.nan infer_memory = np.nan inference_result[model_name][workload] = (infer_time, infer_memory) diff --git a/scripts/conversion_toolkits/README.md b/scripts/conversion_toolkits/README.md index 3ea58dab8b..a348295f2a 100644 --- a/scripts/conversion_toolkits/README.md +++ b/scripts/conversion_toolkits/README.md @@ -39,7 +39,7 @@ Thus, you will need to clone the [electra repository](https://github.com/ZheyuYe and download the checkpoint. The parameters are converted from local checkpoints. By running the following command, you can convert + verify the ELECTRA model with both the discriminator and the generator. -Notice: pleas set up the `--electra_path` with the cloned path if you'd like to directly use `convert_electra.py`. +Notice: please set up the `--electra_path` with the cloned path if you'd like to directly use `convert_electra.py`. ```bash bash convert_electra.sh