Error: Unsupported tensor DataType: DT_INT64, try to modify the model in python to convert the datatype

Hello, I’m trying to convert this TFJS GraphModel:

tfconv.loadGraphModel(
      'https://tfhub.dev/tensorflow/tfjs-model/toxicity/1/default/1',
      { fromTFHub: true })

The conversion works without any issues with the command tfjs_graph_converter --output_format tf_saved_model ./ ./saved/ But, when I try to load the saved model

tf.node.loadSavedModel(this.path)];

I get the error

(node:39361) UnhandledPromiseRejectionWarning: Error: Unsupported tensor DataType: DT_INT64, try to modify the model in python to convert the datatype

[loretoparisi]

@patlevin ah right!!

tfjs_graph_converter --output_format tf_saved_model --compat_mode ./ ./saved/
TensorFlow.js Graph Model Converter

Graph model:    ./
Output:         ./saved/
Target format:  tf_saved_model

Converting.... Done.
Conversion took 1.667s

...
2020-10-05 12:28:55.915996: I tensorflow/cc/saved_model/reader.cc:31] Reading SavedModel from: /Users/loretoparisi/webservice/toxicity_model/saved
2020-10-05 12:28:55.943074: I tensorflow/cc/saved_model/reader.cc:54] Reading meta graph with tags { serve }
2020-10-05 12:28:56.009469: I tensorflow/cc/saved_model/loader.cc:202] Restoring SavedModel bundle.
2020-10-05 12:28:56.010138: I tensorflow/cc/saved_model/loader.cc:212] The specified SavedModel has no variables; no checkpoints were restored. File does not exist: /Users/loretoparisi/webservice/toxicity_model/saved/variables/variables.index
2020-10-05 12:28:56.010284: I tensorflow/cc/saved_model/loader.cc:311] SavedModel load for tags { serve }; Status: success. Took 94285 microseconds.
loadSavedModel OK

Perfect it works! Thank you! 🥇

[patlevin]

@loretoparisi You need to use the compatibility flag:

tfjs_graph_converter --output_format --compat_mode tf_saved_model ./ ./saved/

The compatibility-mode is optional - the default behaviour is to keep all types as-is.