fd6987fafb615427316c0bfac6fdb185273fcfcc,research/object_detection/meta_architectures/center_net_meta_arch.py,CenterNetMetaArch,restore_from_objects,#CenterNetMetaArch#Any#,2954

Before Change

    

    sub_model = self._feature_extractor.get_sub_model(fine_tune_checkpoint_type)
    return {"feature_extractor": sub_model}

  def updates(self):
    raise RuntimeError("This model is intended to be used with model_lib_v2 "
                       "which does not support updates()")

After Change

                         supported_types))

    elif fine_tune_checkpoint_type == "fine_tune":
      feature_extractor_model = tf.train.Checkpoint(
          _feature_extractor=self._feature_extractor)
      return {"model": feature_extractor_model}

    else:
      return {"feature_extractor": self._feature_extractor.get_sub_model(
          fine_tune_checkpoint_type)}

Instances