I have a tensorflow model based Resnet101.
It is a freeze .pb file.
I convert it UFF converter (ver.0.6.3),
It’s input nodes is below:
The probloem is “phase_train” value.
This is used for training or inference on our system.
And it is not problem in TensorFlow framework.
But in Nano with TensorRT,
UFF converter Error is just “not supported Switch layer” because many layers use “phase_train” value.
as you know, for inference mode, just set to false ‘feed_dict’ in Tensorflow session run.
I wonder if “All layer using phase_train like Switch layer must convert Custom layer ?”
or “Another solution like remove-node for inference or set to false value for train and Remake PB file ?”
My tensorflow pb file also has “Dropout” layser.
I found that “Remove dropout layer from frozen-model using python code in tensorflow”.
(link: https://dato.ml/drop-dropout-from-frozen-model )
I wonder if there is such a way “Remove unnecessary switch layer for inference”.