We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cd4236d commit 82bb622Copy full SHA for 82bb622
bioimageio/core/model_adapters/_tensorflow_model_adapter.py
@@ -106,10 +106,10 @@ def _get_network( # pyright: ignore[reportUnknownParameterType]
106
except Exception as e:
107
try:
108
return tf.keras.layers.TFSMLayer(
109
- weight_file, trainable=False, call_endpoint="serve_default"
+ weight_file, trainable=False, call_endpoint="serving_default"
110
) # pyright: ignore[reportUnknownVariableType]
111
- except Exception:
112
- logger.warning(f"error with `call_endpiont='serve': {e}")
+ except Exception as ee:
+ logger.warning(f"error with `call_endpiont='serving_default': {ee}")
113
raise e
114
else:
115
# NOTE in tf1 the model needs to be loaded inside of the session, so we cannot preload the model
0 commit comments