rust icon indicating copy to clipboard operation
rust copied to clipboard

How to load the saved_model pb graph as concrete functions

Open lnshi opened this issue 3 years ago • 1 comments

With TF2 python tf.saved_model.load(export_dir, tags=None, options=None) API, i can load the saved_model as a collection of concrete functions, then i call them like conc_func(**input_tensors) according to different signatures, how can i achieve the same with rust APIs?

Or currently with rust APIs only the model_bundle.session.run(&mut run_args) option is available?

lnshi avatar Dec 23 '21 04:12 lnshi

I do something similar, this may help

https://github.com/Corallus-Caninus/Brains/blob/main/src/lib.rs#L444

the example code also has loading and saving (where I originally got this).

Corallus-Caninus avatar Oct 26 '22 01:10 Corallus-Caninus