我在MNIST上训练了一个ConvNet,并使用simple\u save保存了模型。你知道吗
加载保存的模型时,使用tf.saved_model.loader.load(sess,[tag_constants.SERVING],'/save_folder')
我得到这个错误KeyError: u'ImageProjectiveTransform
保存ConvNet模型的代码
graph = tf.get_default_graph()
from tensorflow.python.saved_model import tag_constants
with graph.as_default():
input_dict = {"x":x,"y":y,"keep_prob":keep_prob,"aug_img":aug_img}
output_dict = {"logits":fc3}
tf.saved_model.simple_save(sess,'/sample_data/',input_dict,output_dict)
我使用tf.contrib.image.rotate
作为扩充技术,但在保存模型时没有将其添加到任何dict中。你知道吗
错误是由此产生的吗?还是有其他问题?你知道吗
目前没有回答
相关问题 更多 >
编程相关推荐