2

我正在使用带有 conda 和 tensorflow 的 python 3,以及以下代码,以创建 tf.keras.models.sequential 并使用 tf.keras.optimizer.Adam 对其进行优化,并得到以下错误:

from tensorflow.python.keras.models import Sequential
from tensorflow.python.keras.datasets import mnist
from tensorflow.python.keras.optimizers import Adam
from tensorflow.python.keras.layers import Dense, Dropout
from siamese import triplet_loss

model = Sequential()
model.add(Dense(units=100, input_shape=(784,), activation="relu"))
model.compile(loss=triplet_loss.TripletLoss.semihard, optimizer=Adam())

(train_x, train_y), (test_x, test_y) = mnist.load_data()
train_x = train_x.reshape((-1, 784)) / 255.0
print(train_x)

ValueError:优化器必须是 tf.train.Optimizer 的实例,而不是

我尝试从 tf.train 导入优化器,但似乎没有找到要导入的内容...

tf 版本是 1.12

谢谢

4

2 回答 2

3

将代码更改为

model = Sequential()
model.add(Dense(units=100, input_shape=(784,), activation="relu"))
model.compile(loss=triplet_loss.TripletLoss.semihard, optimizer=tf.train.AdamOptimizer(learning_rate=0.005))
于 2019-02-15T08:45:28.063 回答
0

这将适用于 Tensorflow 版本 2.x

model = Sequential()
model.add(Dense(units=100, input_shape=(784,), activation="relu"))
model.compile(loss=triplet_loss.TripletLoss.semihard, optimizer=tf.compat.v1.train.AdamOptimizer(learning_rate=0.005))
于 2019-12-21T01:22:14.697 回答