Add tf.py verification printout (#5484)
* Update tf.py with verified confirmation * Update ci-testing.yml * Update ci-testing.yml
This commit is contained in:
1 parent
62d77a1027
commit
06bf8ef7e5
1 file changed
+2
@@ -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()
|
||||
|
||||
Reference in new issue
Block a user