我目前在使用tf.keras.applications中的模型进行训练。同时,还有一个与其一起使用的数据增强层。奇怪的是,在我从applications导入模型后,增强层就不再起作用了。在导入之前它是正常工作的。到底发生了什么?
此外,这只是在TF 2.8.0新版本发布后最近才开始出现的问题。之前都正常工作。
该增强层的代码如下:
data_augmentation = tf.keras.Sequential([
tf.keras.layers.RandomFlip("horizontal_and_vertical"),
tf.keras.layers.RandomRotation(0.5),
])
我正在使用以下方式导入这个模型
base_model = tf.keras.applications.MobileNetV3Small(
input_shape=(75, 50, 3), alpha=1.0,
weights='imagenet', pooling='avg', include_top=False,
dropout_rate=0.1, include_preprocessing=False)
请帮我理解正在发生的事情。您可以在此笔记本上重现代码:https://colab.research.google.com/drive/13Jd3l2CxbvIWQv3Y7CtryOdrv2IdKNxD?usp=sharing。