message: Also add quantization step node to MODEL_VARIABLES collection.
diff: @@ -114,7 +114,9 @@ def CreateOrGetQuantizationStep():
  dtype=dtypes.int64,
  initializer=init_ops.zeros_initializer(),
  trainable=False,
  - collections=[ops.GraphKeys.GLOBAL_VARIABLES])
  + collections=[
  + ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.MODEL_VARIABLES
  + ])
  with g.name_scope(quantization_step_tensor.op.name + '/'):
  # We return the incremented variable tensor. Since this is used in conds
  # for quant_delay and freeze_bn_delay, it will run once per graph
  