glenn-jocher commited on
Commit
06bf8ef
1 Parent(s): 62d77a1

Add tf.py verification printout (#5484)

Browse files

* Update tf.py with verified confirmation

* Update ci-testing.yml

* Update ci-testing.yml

Files changed (1) hide show
  1. models/tf.py +2 -0
models/tf.py CHANGED
@@ -441,6 +441,8 @@ def run(weights=ROOT / 'yolov5s.pt', # weights path
441
  keras_model = keras.Model(inputs=im, outputs=tf_model.predict(im))
442
  keras_model.summary()
443
 
 
 
444
 
445
  def parse_opt():
446
  parser = argparse.ArgumentParser()
 
441
  keras_model = keras.Model(inputs=im, outputs=tf_model.predict(im))
442
  keras_model.summary()
443
 
444
+ LOGGER.info('PyTorch, TensorFlow and Keras models successfully verified.\nUse export.py for TF model export.')
445
+
446
 
447
  def parse_opt():
448
  parser = argparse.ArgumentParser()