debug value

official/vision/beta/projects/yolo/dataloaders/classification_input.py

@@ -27,7 +27,7 @@ STDDEV_RGB = (0.229 * 255, 0.224 * 255, 0.225 * 255)
 class Decoder(decoder.Decoder):
   """A tf.Example decoder for classification task."""
   def __init__(self):
-    print("decoder ahs been init")
+    tf.print("decoder ahs been init")
     return
 
   def decode(self, serialized_example):

official/vision/beta/tasks/image_classification.py

@@ -54,10 +54,10 @@ class ImageClassificationTask(base_task.Task):
     input_size = self.task_config.model.input_size
 
     if params.tfds_name != None: 
-      print("i am here for training using tfds")
+      tf.print("i am here for training using tfds")
       decoder = cli.Decoder()
     else:
-      print("i am here for regular input")
+      tf.print("i am here for regular input")
       decoder = classification_input.Decoder()
 
     parser = classification_input.Parser(