From 653417ae4bdd6b166006e2aa995be7868ac04447 Mon Sep 17 00:00:00 2001 From: Sergii Dymchenko Date: Mon, 8 Jun 2020 13:02:12 -0700 Subject: [PATCH] Fix scaler->scalar typo. (#4142) --- orttraining/orttraining/python/ort_trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/orttraining/orttraining/python/ort_trainer.py b/orttraining/orttraining/python/ort_trainer.py index b9d7be4c93..e8a3b637dd 100644 --- a/orttraining/orttraining/python/ort_trainer.py +++ b/orttraining/orttraining/python/ort_trainer.py @@ -971,7 +971,8 @@ class ORTTrainer(): model.graph.output[0].type.tensor_type.elem_type != onnx.TensorProto().BFLOAT16: raise RuntimeError("the first output of a model to run with fully optimized ORT backend must be float types.") if len(model.graph.output[0].type.tensor_type.shape.dim) != 0: - raise RuntimeError("the first output of a model to run with fully optimized ORT backend must be a scaler.") + raise RuntimeError( + "the first output of a model to run with fully optimized ORT backend assumed to be loss and must be a scalar.") class LossScaler(): def __init__(self, loss_scale_input_name, is_dynamic_scale,