From 24db8cc329ad81bc4ca9e716ae3e055c17d0f2e2 Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Sun, 7 Feb 2021 17:54:20 -0800 Subject: [PATCH] Can't mix --fp16 and --device cpu (#10041) --- examples/seq2seq/run_eval.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/examples/seq2seq/run_eval.py b/examples/seq2seq/run_eval.py index 739efc531a..e21f57c1c6 100755 --- a/examples/seq2seq/run_eval.py +++ b/examples/seq2seq/run_eval.py @@ -132,8 +132,14 @@ def run_generate(verbose=True): if args.n_obs > 0: examples = examples[: args.n_obs] Path(args.save_path).parent.mkdir(exist_ok=True) + if args.reference_path is None and Path(args.score_path).exists(): warnings.warn(f"score_path {args.score_path} will be overwritten unless you type ctrl-c.") + + if args.device == "cpu" and args.fp16: + # this mix leads to RuntimeError: "threshold_cpu" not implemented for 'Half' + raise ValueError("Can't mix --fp16 and --device cpu") + runtime_metrics = generate_summaries_or_translations( examples, args.save_path,