debug value

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

@@ -26,6 +26,9 @@ 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")
+    return
 
   def decode(self, serialized_example):
     sample_dict = {

official/vision/beta/tasks/image_classification.py

@@ -54,9 +54,12 @@ 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")
       decoder = cli.Decoder()
     else:
+      print("i am here for regular input")
       decoder = classification_input.Decoder()
+
     parser = classification_input.Parser(
         output_size=input_size[:2],
         num_classes=num_classes,