machine-learning - 如何修复pycharm中导入tf包的这个错误

Question

我在导入张量流包时出错。错误是 ImportError: No module named 'tf.saved_model'; 'tf' 不是一个包

这是初始模型，我需要将此模型保存在我的数据集上，以避免一次又一次地训练模型

ImportError：没有名为“tf.saved_model”的模块；'tf' 不是一个包

以上是运行程序时发生的错误

标签： machine-learningdeep-learningcomputer-vision

Answer 1

In your case:
    import tensorflow as tf

    export_dir = 'your_dir'

    builder = tf.saved_model.builder.SavedModelBuilder(export_dir)

    builder.add_meta_graph_and_variables(
                sess, [tf.saved_model.tag_constants.SERVING],
                signature_def_map={
                    tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
                        prediction_signature
                }
            )
    builder.save()