internal change

PiperOrigin-RevId: 622868665

Author: TensorFlow Hub Authors

examples/image_retraining/retrain.py

@@ -900,7 +900,7 @@ def build_eval_session(module_spec, class_count):
 
 
 def save_graph_to_file(graph_file_name, module_spec, class_count):
-  """Saves an graph to file, creating a valid quantized one if necessary."""
+  """Saves a graph to file, creating a valid quantized one if necessary."""
   sess, _, _, _, _, _ = build_eval_session(module_spec, class_count)
   graph = sess.graph
 
@@ -1109,9 +1109,6 @@ def main(_):
                      datetime.now(), i, train_accuracy * 100)
         logging.info('%s: Step %d: Cross entropy = %f',
                      datetime.now(), i, cross_entropy_value)
-        # TODO: Make this use an eval graph, to avoid quantization
-        # moving averages being updated by the validation set, though in
-        # practice this makes a negligable difference.
         validation_bottlenecks, validation_ground_truth, _ = (
             get_random_cached_bottlenecks(
                 sess, image_lists, FLAGS.validation_batch_size, 'validation',

tensorflow_hub/keras_layer.py

@@ -338,7 +338,6 @@ def get_config(self):
     if not isinstance(self._handle, str):
       # Need to raise this type in order for tf.saved_model.save() to fall back
       # to not using config, instead of crashing.
-      # TODO(b/134528831): Reconsider the usability implications.
       raise NotImplementedError(
           "Can only generate a valid config for `hub.KerasLayer(handle, ...)`"
           "that uses a string `handle`.\n\n"

tensorflow_hub/keras_layer_test.py

@@ -602,9 +602,6 @@ def testResaveWithMixedPrecision(self, save_from_keras):
                     "only available in TF 2.4 or above")
     export_dir1 = os.path.join(self.get_temp_dir(), "mixed-precision")
     export_dir2 = os.path.join(self.get_temp_dir(), "mixed-precision2")
-    # TODO(b/193472950): Currently, KerasLayer only works with mixed precision
-    # when the model takes non-floating point inputs, which is why an embedding
-    # model is used in this test.
     _save_2d_text_embedding(export_dir1, save_from_keras=save_from_keras)
     try:
       tf_keras_v2.mixed_precision.set_global_policy("mixed_float16")

tensorflow_hub/resolver.py

@@ -401,9 +401,6 @@ def atomic_download(handle,
               tf.errors.UnimplementedError):
         raise
       # All other errors are retried.
-      # TODO(b/144424849): Retrying an AlreadyExistsError from the atomic write
-      # should be good enough, but see discussion about misc filesystem types.
-      # TODO(b/144475403): How atomic is the overwrite=False check?
       except tf.errors.OpError:
         pass
 
@@ -542,4 +539,3 @@ def _maybe_disable_cert_validation(self):
       self._context.check_hostname = False
       self._context.verify_mode = ssl.CERT_NONE
       logging.warning("Disabled certificate validation for resolving handles.")
-