Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
sxjscience committed Aug 10, 2020
1 parent a43e65b commit b1f5955
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion scripts/benchmarks/benchmark_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
2 changes: 1 addition & 1 deletion scripts/conversion_toolkits/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit b1f5955

Please sign in to comment.