I save a pb model.the model define as below:
def resnet_18(features, params, is_training, reuse_variables=None, aux_features=None):
…
features = tf.layers.batch_normalization(features, momentum=params.batchnorm_momentum, training=is_training, name=“conv0_bn”)
…
I save the pb when predict and can restore the pb from tf and run success.
I compile the pb with tvm is successful,too.
But build error TVMError: Check failed: checked_type_.defined(): internal error: the type checker has not populated the checked_type field for CallNode.
I think is the is_training’s problem.
Any suggestion?Thanks