From 06bf8ef7e5a403ef7d50b4962a5d9851e20e689b Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 3 Nov 2021 19:04:51 +0100
Subject: [PATCH] Add tf.py verification printout (#5484)

* Update tf.py with verified confirmation

* Update ci-testing.yml

* Update ci-testing.yml
---
 models/tf.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/tf.py b/models/tf.py
index 61494b4a1fae..6c07410e03a5 100644
--- a/models/tf.py
+++ b/models/tf.py
@@ -441,6 +441,8 @@ def run(weights=ROOT / 'yolov5s.pt',  # weights path
     keras_model = keras.Model(inputs=im, outputs=tf_model.predict(im))
     keras_model.summary()
 
+    LOGGER.info('PyTorch, TensorFlow and Keras models successfully verified.\nUse export.py for TF model export.')
+
 
 def parse_opt():
     parser = argparse.ArgumentParser()