s/training_op/train_op/ (#10028)
This commit is contained in:
parent
e47e3b9826
commit
47bcf2919f
@ -132,7 +132,7 @@ class ModelFnOps(
|
||||
# Validate train_op.
|
||||
if train_op is None:
|
||||
if mode == ModeKeys.TRAIN:
|
||||
raise ValueError('Missing training_op.')
|
||||
raise ValueError('Missing train_op.')
|
||||
elif not isinstance(train_op, ops.Operation):
|
||||
# TODO(ptucker): Should this be allowed? Consider raising error.
|
||||
train_op = ops.convert_to_tensor(train_op).op
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user