search

naturomics / CapsNet-Tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules
Apache License 2.0
3.8k stars 1.16k forks source link

How to save model in SavedModel or frozen_graph format? #85

Open Azuresonance opened 3 years ago

Azuresonance commented 3 years ago

Is there a way to save this model in SavedModel or frozen_graph format? I am trying to do inference using Tensorrt, which only supports these two formats.