Commit c96825a4 authored by Vishnu Banna's avatar Vishnu Banna
Browse files

Merge branch 'yolo' of https://github.com/PurdueCAM2Project/tf-models into yolo

parents d791ef44 9d58e701
......@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment