【发布时间】:2021-03-06 07:48:24
【问题描述】:
这是代码:
with strategy.scope():
model2 = tf.keras.applications.VGG16(
include_top=True,
weights=None,
input_tensor=None,
input_shape=(32, 32, 3),
pooling=None,
classes=10,
classifier_activation="relu",
)
model2.compile(optimizer='adam',
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
metrics=['accuracy'])
history = model2.fit(
train_images, train_labels,epochs=10,
validation_data=(test_images, test_labels)
)
我正在尝试从头开始训练 VGG16,因此不导入它们的权重(因此 weights=None)
我还尝试了一个我自己创建的模型,具有相同的超参数,效果很好。
这里可能有什么问题?
非常感谢任何帮助
Heres完整代码
【问题讨论】:
标签: deep-learning tensor