Description
Ask the question
I am trying to load a TensorFlow SavedModel using the TensorFlowSavedModelExternalModel class but getting an error related to the model signature (anonymized stack trace below). My model has two signatures, "basic" and "embeddings". I can't find a way in the Tribuo interface to specify which signature I want to use. Is it possible to specify, or can TensorFlowSavedModelExternalModel only be used for classes which provide a serving_default signature? Thanks!
Is your question about a specific ML algorithm or approach?
No
Is your question about a specific Tribuo class?
TensorFlowSavedModelExternalModel
System details
- Tribuo version
4.3.1 - Java version (if appropriate)
21 - OS/Architecture (if appropriate) Windows
11
Additional context
Stack trace with path replaced with {path_to_model} for privacy:
2025-02-11 09:46:53.530519: I external/org_tensorflow/tensorflow/cc/saved_model/reader.cc:43] Reading SavedModel from: {path_to_model}
2025-02-11 09:46:53.584789: I external/org_tensorflow/tensorflow/cc/saved_model/reader.cc:107] Reading meta graph with tags { serve }
2025-02-11 09:46:53.584823: I external/org_tensorflow/tensorflow/cc/saved_model/reader.cc:148] Reading SavedModel debug info (if present) from: {path_to_model}
2025-02-11 09:46:53.585820: I external/org_tensorflow/tensorflow/core/platform/cpu_feature_guard.cc:151] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2025-02-11 09:46:53.798251: I external/org_tensorflow/tensorflow/cc/saved_model/loader.cc:228] Restoring SavedModel bundle.
2025-02-11 09:46:54.295126: I external/org_tensorflow/tensorflow/cc/saved_model/loader.cc:212] Running initialization op on SavedModel bundle at path: {path_to_model}
2025-02-11 09:46:54.578628: I external/org_tensorflow/tensorflow/cc/saved_model/loader.cc:301] SavedModel load for tags { serve }; Status: success: OK. Took 1048104 microseconds.
Error loading model: Cannot elect a default function for this model