TensorFlow 模型建立与训练

技术分享 手册留言
17

使用 cats_vs_dogs 数据集，按照下面的模型结构训练 CNN，10 个 epochs 的训练时间将近 3 个小时，请问这个正常吗？

model = tf.keras.Sequential ([
    tf.keras.layers.Conv2D (32, 3, activation='relu', input_shape=(256, 256, 3)),
    tf.keras.layers.MaxPooling2D (),
    tf.keras.layers.Conv2D (32, 5, activation='relu'),
    tf.keras.layers.MaxPooling2D (),
    tf.keras.layers.Flatten (),
    tf.keras.layers.Dense (64, activation='relu'),
    tf.keras.layers.Dense (2, activation='softmax')
])