saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-25 19:48:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix scaler->scalar typo. (#4142)

Browse source
This commit is contained in:
Sergii Dymchenko 2020-06-08 13:02:12 -07:00 committed by GitHub
parent 6bbd18efd0
commit 653417ae4b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
orttraining/orttraining/python/ort_trainer.py
View file

@ -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,