一般出现此类问题的原因是包的更新导致有些用法发生了变化,因此在tensorflow中调用optimizer需要通过tf.keras调用。

self.opt = tf.optimizers.Adam(learning_rate=self.lr)中的tf后面加个keras,
变成self.opt = tf.keras.optimizers.Adam(learning_rate=self.lr)

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐