在Kerass Tensorflow 2.5中使用自定义优化器

我使用adam和sgd优化器，使用以下代码时没有问题：

model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])

但是，当我尝试创建并传递优化器时，我收到错误：

opt = tf.keras.optimizers.SGD(learning_rate=0.01, momentum=0.9)
model.compile(loss='categorical_crossentropy', optimizer=opt, metrics=['accuracy'])

我收到以下错误：

ValueError: Could not interpret optimizer identifier: <tensorflow.python.keras.optimizer_v2.gradient_descent.SGD object at 0x000001AACFDAE460>

ValueError:无法解释优化器标识符：

您需要包含包含导入的完整示例。