TOCO failed Batch normalization resolution requires that mean, multiplier and offset arrays be constant.
I'm using tensorflow and have created a custom model. The training process is ok, but when I want to freeze the graph and convert it into tflite, I'm getting this error:
TOCO failed Batch normalization resolution requires that mean, multiplier and offset arrays be constant.
I can freeze the graph separately, but I can't convert it to tflite. Here is the code of freezing the graph.
g = tf.get_default_graph()
graph_def = g.as_graph_def()
tf.train.write_graph(graph_def, "./model", 'graph.pb', as_text=False)
saver = tf.train.Saver()
saver.save(sess, os.path.join(FLAGS.checkpoint_dir, 'model.ckpt'))