From 9940348d954602acbaa0adb5d7f58c1bd5f0d36a Mon Sep 17 00:00:00 2001 From: felix Date: Tue, 1 Oct 2024 14:19:33 +0200 Subject: [PATCH] update --- tests/tensorflow/test_models_recognition_tf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tensorflow/test_models_recognition_tf.py b/tests/tensorflow/test_models_recognition_tf.py index bdc8441c8f..162c446d35 100644 --- a/tests/tensorflow/test_models_recognition_tf.py +++ b/tests/tensorflow/test_models_recognition_tf.py @@ -194,7 +194,7 @@ def test_recognition_zoo_error(): def test_models_onnx_export(arch_name, input_shape): # Model batch_size = 2 - tf.backend.clear_session() + tf.keras.backend.clear_session() model = recognition.__dict__[arch_name](pretrained=True, exportable=True, input_shape=input_shape) # SAR, MASTER, ViTSTR export currently only available with constant batch size if arch_name in ["sar_resnet31", "master", "vitstr_small", "parseq"]: