Error-tf.function-decorated function tried to create variables on non-first call

最新推荐文章于 2023-03-30 11:09:23 发布

原创最新推荐文章于 2023-03-30 11:09:23 发布 · 937 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #人工智能 #python

Code-error 同时被 2 个专栏收录

41 篇文章

订阅专栏

Tensorflow

19 篇文章

订阅专栏

在TensorFlow2中遇到模型保存时报错，原因是尝试在非首次调用时创建变量。解决方法是在`__init__`函数中预先定义好需要的层，例如在初始化时创建`Conv2D`层，然后在`call`函数中根据条件调用。这样可以确保变量在正确的时间被创建，避免保存模型时出错。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ValueError: tf.function-decorated function tried to create variables on non-first call

在tensorflow2中如下保存模型的总是报错,很是头疼:

model.save(weight_file)

错误的位置定格在:

out_data = tf.keras.layers.Conv2D(filters, [3,3])(concat_data)

最终找到原因,需要先在__init__函数体中定义好,才能在call中进行调用:

    def __init__(self,**kwargs):
        super(Up_sampling, self).__init__(**kwargs)
        self.Conv2D_256 = tf.keras.layers.Conv2D(256, [3, 3])
    def call(self, input,filters):
        if filters==256:
            out_data = self.Conv2D_256(input)

这样就很好的解决问题了,错误消失!