You could do something like: for layer in base_model.layers: if isinstance(layer ,tf.keras.layers.BatchNormalization): layer.trainable=False.
確定! 回上一頁